MlirPassthroughOp

کلاس نهایی عمومی MlirPassthroughOp

یک محاسبات دلخواه 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
ایجاد ( دامنه دامنه ، Iterable< Operand <?>> ورودی، String mlirModule، List<Class<?>> Toutputs)
روش کارخانه برای ایجاد کلاسی که یک عملیات MlirPassthroughOp جدید را بسته بندی می کند.
Iterator< Operand <Object>>
فهرست< خروجی <?>>

روش های ارثی

روش های عمومی

ایجاد عمومی ثابت MlirPassthroughOp ( دامنه دامنه ، Iterable< Operand <?>> ورودی ها، String mlirModule، List<Class<?>> Toutputs)

روش کارخانه برای ایجاد کلاسی که یک عملیات MlirPassthroughOp جدید را بسته بندی می کند.

پارامترها
دامنه محدوده فعلی
برمی گرداند
  • یک نمونه جدید از MlirPassthroughOp

عمومی Iterator< Operand <Object>> تکرار کننده ()

فهرست عمومی< خروجی <?>> خروجی ها ()