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> } '''
Genel Yöntemler
statik MlirGeçişOp | |
Yineleyici< İşlenen <Nesne>> | yineleyici () |
Liste< Çıkış <?>> | çıktılar () |
Kalıtsal Yöntemler
Genel Yöntemler
public static MlirPassthroughOp create ( Kapsam kapsamı, Yinelenebilir< İşlenen <?>> girişleri, String mlirModule, List<Class<?>> Çıkışları)
Yeni bir MlirPassthroughOp işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|
İadeler
- MlirPassthroughOp'un yeni bir örneği