MlirPassthroughOp

clase final pública MlirPassthroughOp

Envuelve un cálculo MLIR arbitrario expresado como un módulo con una función main ().

Esta operación no tiene un kernel asociado y no está diseñada para ejecutarse en una sesión regular de TensorFlow. En su lugar, está destinado a ser utilizado para pruebas o para casos especiales en los que un usuario tiene la intención de pasar un cálculo MLIR personalizado a través de un gráfico de TensorFlow con la intención de que las herramientas personalizadas lo procesen en sentido descendente (cuando se dirige a un entorno diferente, como TensorFlow lite, por ejemplo). Se espera que el módulo MLIR tenga una función main () que se utilizará como punto de entrada. Las entradas a las operaciones se pasarán como argumento a la función main () y los valores devueltos de la función principal se asignarán a las salidas. Uso de ejemplo:

 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>
 
 '''

    
    

    

Métodos públicos

static MlirPassthroughOp	create ( Scope scope, Iterable < Operand <? >> entradas, String mlirModule, List <Class <? >> Toutputs) Método de fábrica para crear una clase que envuelva una nueva operación MlirPassthroughOp.
Iterador < Operando <Objeto>>	iterador ()
Lista < Salida <? >>	salidas ()

Métodos heredados

De la clase org.tensorflow.op.PrimitiveOp

booleano final	es igual a (Objeto obj)
int final	hashCode ()
Operación	op () Devuelve la Operation subyacente
cadena final	toString ()

De la clase java.lang.Object

booleano	es igual a (Objeto arg0)
Clase final <?>	getClass ()
En t	hashCode ()
vacío final	notificar ()
vacío final	notificar a todos ()
Cuerda	toString ()
vacío final	esperar (largo arg0, int arg1)
vacío final	espera (largo arg0)
vacío final	esperar ()

Desde la interfaz java.lang.Iterable

vacío abstracto	forEach (consumidor <? super T> arg0)
Iterador abstracto < Operando <Objeto>>	iterador ()
spliterator abstracto < operando <Objeto>>	spliterator ()

Métodos públicos

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

Método de fábrica para crear una clase que envuelva una nueva operación MlirPassthroughOp.

Parámetros

alcance	alcance actual

Devoluciones

• una nueva instancia de MlirPassthroughOp

iterador público < operando <Objeto>> iterador ()

Lista pública < Salida <? >> salidas ()