Membungkus komputasi MLIR arbitrer yang dinyatakan sebagai modul dengan fungsi main().
Operasi ini tidak memiliki kernel terkait dan tidak dimaksudkan untuk dijalankan dalam sesi TensorFlow reguler. Sebaliknya, ini dimaksudkan untuk digunakan untuk pengujian atau untuk kasus khusus ketika pengguna bermaksud meneruskan komputasi MLIR khusus melalui grafik TensorFlow dengan tujuan agar alat khusus memprosesnya secara hilir (saat menargetkan lingkungan yang berbeda, seperti TensorFlow lite misalnya). Modul MLIR diharapkan memiliki fungsi main() yang akan digunakan sebagai titik masuk. Masukan ke operasi akan diteruskan sebagai argumen ke fungsi main() dan nilai yang dikembalikan dari fungsi utama dipetakan ke keluaran. Contoh penggunaan:
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>
'''
Konstanta
Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Publik
MlirPassthroughOp statis | |
Iterator< Operan < TType >> | pengulangan () |
Daftar< Keluaran <?>> | keluaran () |
Metode Warisan
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Metode Publik
public static MlirPassthroughOp buat ( Lingkup cakupan , Iterable< Operand <?>> input, String mlirModule, Daftar<Kelas<? extends TType >> Toutputs)
Metode pabrik untuk membuat kelas yang membungkus operasi MlirPassthroughOp baru.
Parameter
cakupan | ruang lingkup saat ini |
---|
Kembali
- contoh baru MlirPassthroughOp