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:
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>
'''
Sabitler
Sicim | OP_NAME | Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı |
Genel Yöntemler
statik MlirGeçişOp | |
Yineleyici< İşlenen < TType >> | yineleyici () |
Liste< Çıkış <?>> | çıktılar () |
Kalıtsal Yöntemler
Sabitler
genel statik son Dize OP_NAME
Bu operasyonun TensorFlow çekirdek motoru tarafından bilinen adı
Genel Yöntemler
public static MlirPassthroughOp create ( Kapsam kapsamı, Yinelenebilir< İşlenen <?>> girişleri, String mlirModule, List<Class<? extends TType >> Toutputs)
Yeni bir MlirPassthroughOp işlemini saran bir sınıf oluşturmaya yönelik fabrika yöntemi.
Parametreler
kapsam | mevcut kapsam |
---|
İade
- MlirPassthroughOp'un yeni bir örneği