MlirPassthroughOp

public final class MlirPassthroughOp

Avvolge un calcolo MLIR arbitrario espresso come un modulo con una funzione main ().

Questa operazione non ha un kernel associato e non deve essere eseguita in una normale sessione TensorFlow. È invece pensato per 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 TensorFlow lite per esempio). Il modulo MLIR dovrebbe avere una funzione main () che verrà utilizzata come punto di ingresso. Gli input per le operazioni verranno passati come argomenti alla funzione main () ei valori restituiti dalla funzione main 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

static MlirPassthroughOp	create ( Scope scope, Iterable < Operand <? >> input, String mlirModule, List <Class <? >> Toutputs) Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione MlirPassthroughOp.
Iteratore < Operando <Oggetto>>	iteratore ()
Elenco < Uscita <? >>	uscite ()

Metodi ereditati

Dalla classe org.tensorflow.op.PrimitiveOp

booleano finale	è uguale a (Object obj)
final int	hashCode ()
Operazione	op () Restituisce l' Operation sottostante
stringa finale	toString ()

Dalla classe java.lang.Object

booleano	è uguale a (Oggetto arg0)
classe finale <?>	getClass ()
int	hashCode ()
vuoto finale	notifica ()
vuoto finale	notifyAll ()
Corda	toString ()
vuoto finale	wait (long arg0, int arg1)
vuoto finale	aspetta (lungo arg0)
vuoto finale	aspetta ()

Dall'interfaccia java.lang.Iterable

vuoto astratto	forEach (Consumer <? super T> arg0)
Iteratore astratto < Operando <Oggetto>>	iteratore ()
abstract Spliterator < Operando <Object>>	spliterator ()

Metodi pubblici

public static MlirPassthroughOp create ( Scope scope, Iterable < Operand <? >> input, String mlirModule, List <Class <? >> Toutputs)

Metodo Factory per creare una classe che esegue il wrapping di una nuova operazione MlirPassthroughOp.

Parametri

scopo	ambito attuale

ritorna

• una nuova istanza di MlirPassthroughOp

iteratore pubblico < Operando <Oggetto>> iteratore ()

public List < Output <? >> output ()