یک محاسبات دلخواه MLIR را که به صورت یک ماژول با تابع ()main بیان شده است، می پیچد.
این عملیات هسته مرتبطی ندارد و قرار نیست در یک جلسه TensorFlow معمولی اجرا شود. در عوض در نظر گرفته شده است که برای آزمایش یا برای موارد خاصی استفاده شود که در آن کاربر قصد دارد محاسبات MLIR سفارشی را از طریق یک نمودار TensorFlow با هدف پردازش ابزار سفارشی آن در پایین دست (هنگامی که یک محیط متفاوت را هدف قرار می دهد، مانند TensorFlow lite) منتقل کند. انتظار می رود ماژول MLIR دارای یک تابع main() باشد که به عنوان نقطه ورودی استفاده می شود. ورودیهای عملیات بهعنوان آرگومان به تابع main() ارسال میشوند و مقادیر بازگشتی تابع main به خروجیها نگاشت میشوند. مثال استفاده:
{@code 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> } '''
روش های عمومی
استاتیک MlirPassthroughOp | |
Iterator< Operand <Object>> | اشاره گر () |
فهرست< خروجی <?>> | خروجی ها () |
روش های ارثی
روش های عمومی
ایجاد عمومی ثابت MlirPassthroughOp ( دامنه دامنه ، Iterable< Operand <?>> ورودی ها، String mlirModule، List<Class<?>> Toutputs)
روش کارخانه برای ایجاد کلاسی که یک عملیات MlirPassthroughOp جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|
برمی گرداند
- یک نمونه جدید از MlirPassthroughOp