MlirPassthroughOp

يلتف حساب MLIR التعسفي معبرًا عنه كوحدة مع وظيفة main().

لا تحتوي هذه العملية على نواة مرتبطة وليس المقصود تنفيذها في جلسة TensorFlow عادية. بدلاً من ذلك، تم تصميمه للاستخدام للاختبار أو لحالة خاصة حيث ينوي المستخدم تمرير حساب MLIR مخصص من خلال رسم بياني TensorFlow بقصد الحصول على أدوات مخصصة لمعالجته في اتجاه مجرى النهر (عند استهداف بيئة مختلفة، مثل TensorFlow lite على سبيل المثال). من المتوقع أن تحتوي وحدة MLIR على وظيفة 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>
 }
 '''