MlirPassthroughOp

lezione finale pubblica MlirPassthroughOp

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:

 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	create ( Scope scope, Iterable< Operand <?>> input, String mlirModule, List<Class<?>> Toutputs) Metodo factory per creare una classe che racchiude una nuova operazione MlirPassthroughOp.
Iteratore< Operando <Oggetto>>	iteratore ()
Elenco< Uscita <?>>	uscite ()

Metodi ereditati

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Oggetto oggetto)
finale int	codice hash ()
Operazione	operazione () Restituisce l' Operation sottostante
stringa finale	accordare ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
Classe finale<?>	getClass ()
int	codice hash ()
vuoto finale	notificare ()
vuoto finale	notificaTutti ()
Corda	accordare ()
vuoto finale	attendere (lungo arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	Aspettare ()

Dall'interfaccia java.lang.Iterable

vuoto astratto	forEach (Consumatore<? super T> arg0)
abstract Iteratore< Operando <Oggetto>>	iteratore ()
abstract Spliterator< Operando <Oggetto>>	divisore ()

Metodi pubblici

creazione pubblica statica MlirPassthroughOp (ambito ambito , input Iterable< Operand <?>>, String mlirModule, List<Class<?>> Toutputs)

Metodo factory per creare una classe che racchiude una nuova operazione MlirPassthroughOp.

Parametri

scopo	ambito attuale

ritorna

• una nuova istanza di MlirPassthroughOp

public Iterator< Operando <Oggetto>> iteratore ()

elenco pubblico< Output <?>> uscite ()