Racchiude un calcolo MLIR arbitrario espresso come modulo con una funzione main().
Questa operazione non ha un kernel associato e non è destinata ad essere eseguita in una normale sessione di TensorFlow. È invece destinato a essere utilizzato per test o per casi speciali in cui un utente intende passare il calcolo MLIR personalizzato attraverso un grafico TensorFlow con l'intento di avere strumenti personalizzati che lo elaborano a valle (quando si prende di mira un ambiente diverso, come ad esempio TensorFlow lite). Si prevede che il modulo MLIR disponga di una funzione main() che verrà utilizzata come punto di ingresso. Gli input per le operazioni verranno passati come argomento alla funzione main() e i valori restituiti della funzione main verranno mappati sugli output. Utilizzo di esempio:
{@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> } '''
Metodi pubblici
MlirPassthrough staticoOp | |
Iteratore< Operando <Oggetto>> | iteratore () |
Elenco< Uscita <?>> | uscite () |
Metodi ereditati
Metodi pubblici
creazione statica pubblica MlirPassthroughOp ( ambito ambito , input Iterable< Operand <?>>, String mlirModule, List<Class<?>> Toutputs)
Metodo factory per creare una classe che racchiude una nuova operazione MlirPassthroughOp.
Parametri
portata | ambito attuale |
---|
Ritorni
- una nuova istanza di MlirPassthroughOp