MlirPassthroughOp

Main() işlevine sahip bir modül olarak ifade edilen rastgele bir MLIR hesaplamasını sarar.

Bu işlemin ilişkili bir çekirdeği yoktur ve normal bir TensorFlow oturumunda yürütülmesi amaçlanmamıştır. Bunun yerine, test amacıyla veya bir kullanıcının özel MLIR hesaplamasını bir TensorFlow grafiği aracılığıyla, özel araçların bunu aşağı yönde işlemesini sağlamak amacıyla (örneğin TensorFlow lite gibi farklı bir ortamı hedeflerken) geçirmeyi planladığı özel durumlar için kullanılması amaçlanmıştır. MLIR modülünün giriş noktası olarak kullanılacak bir main() fonksiyonuna sahip olması beklenmektedir. İşlemlerin girdileri, main() işlevine argüman olarak iletilecek ve ana işlevin döndürülen değerleri, çıktılarla eşleştirilecektir. Örnek kullanım:

{@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>
 }
 '''