MlirPassthroughOp

kelas akhir publik MlirPassthroughOp

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
buat ( Lingkup lingkup, input Iterable< Operand <?>>, String mlirModule, Daftar<Kelas<? extends TType >> Toutputs)
Metode pabrik untuk membuat kelas yang membungkus operasi MlirPassthroughOp baru.
Iterator< Operan < TType >>
Daftar< Keluaran <?>>

Metode Warisan

Konstanta

String akhir statis publik OP_NAME

Nama operasi ini dikenal dengan mesin inti TensorFlow

Nilai Konstan: "MlirPassthroughOp"

Metode Publik

pembuatan MlirPassthroughOp statis publik ( Lingkup lingkup, input Iterable< Operand <?>>, 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

Iterator publik< Operan < TType >> iterator ()

Daftar publik< Keluaran <?>> keluaran ()