یک محاسبات دلخواه MLIR را که به صورت یک ماژول با تابع ()main بیان شده است، می پیچد.
این عملیات هسته مرتبطی ندارد و قرار نیست در یک جلسه TensorFlow معمولی اجرا شود. در عوض در نظر گرفته شده است که برای آزمایش یا برای موارد خاصی استفاده شود که در آن کاربر قصد دارد محاسبات MLIR سفارشی را از طریق یک نمودار TensorFlow با هدف پردازش ابزار سفارشی آن در پایین دست (هنگامی که یک محیط متفاوت را هدف قرار می دهد، مانند TensorFlow lite) منتقل کند. انتظار می رود ماژول MLIR دارای یک تابع main() باشد که به عنوان نقطه ورودی استفاده می شود. ورودیهای عملیات بهعنوان آرگومان به تابع main() ارسال میشوند و مقادیر بازگشتی تابع main به خروجیها نگاشت میشوند. مثال استفاده:
import tensorflow as tf
from tensorflow.compiler.mlir.tensorflow.gen_mlir_passthrough_op import mlir_passthrough_op
mlir_module = '''python
func @main(%arg0 : tensor<10xf32>, %arg1 : tensor<10xf32>) -> tensor<10x10xf32> {
%add = "magic.op"(%arg0, %arg1) : (tensor<10xf32>, tensor<10xf32>) -> tensor<10x10xf32>
return %ret : tensor<10x10xf32>
'''
ثابت ها
رشته | OP_NAME | نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود |
روش های عمومی
استاتیک MlirPassthroughOp | |
Iterator< Operand < TType >> | اشاره گر () |
فهرست< خروجی <?>> | خروجی ها () |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
روش های عمومی
ایجاد عمومی ثابت MlirPassthroughOp ( دامنه دامنه ، Iterable< Operand <?>> ورودی، String mlirModule، List<Class<? گسترش TType >> Toutputs)
روش کارخانه برای ایجاد کلاسی که یک عملیات MlirPassthroughOp جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|
برمی گرداند
- یک نمونه جدید از MlirPassthroughOp