MlirPassthroughOp

public final class MlirPassthroughOp

Envolve uma computação MLIR arbitrária expressa como um módulo com uma função main ().

Esta operação não tem um kernel associado e não se destina a ser executada em uma sessão regular do TensorFlow. Em vez disso, destina-se a ser usado para teste ou para casos especiais em que um usuário pretende passar o cálculo MLIR personalizado por meio de um gráfico do TensorFlow com a intenção de ter ferramentas personalizadas processando-o downstream (ao direcionar para um ambiente diferente, como o TensorFlow Lite, por exemplo). Espera-se que o módulo MLIR tenha uma função main () que será usada como um ponto de entrada. As entradas para as operações serão passadas como argumento para a função main () e os valores retornados da função principal mapeados para as saídas. Exemplo de utilização:

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

estática MlirPassthroughOp	criar ( Scope escopo, Iterable < Operando <? >> entradas, String mlirModule, List <Class <? >> Toutputs) Método de fábrica para criar uma classe envolvendo uma nova operação MlirPassthroughOp.
Iterator < Operando <Object >>	iteração ()
List < Output <? >>	saídas ()

Métodos herdados

De classe org.tensorflow.op.PrimitiveOp

final booleano	é igual a (obj Object)
final int	hashCode ()
Operação	op () Retorna o subjacente Operation
corda final	toString ()

Da classe java.lang.Object

boleano	equals (arg0 Object)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notifyAll ()
Fragmento	toString ()
vazio final	wait (longa arg0, int arg1)
vazio final	wait (arg0 de comprimento)
vazio final	wait ()

Da interface java.lang.Iterable

vazio abstrato	foreach (Consumer <? Super T> arg0)
abstrato Iterator < Operando <Object >>	iteração ()
Spliterator abstrato < Operando <Object >>	spliterator ()

Métodos Públicos

public static MlirPassthroughOp criar ( Scope escopo, Iterable < Operando <? >> entradas, String mlirModule, List <Class <? >> Toutputs)

Método de fábrica para criar uma classe envolvendo uma nova operação MlirPassthroughOp.

Parâmetros

alcance	escopo atual

Devoluções

• uma nova instância de MlirPassthroughOp

pública Iterator < Operando <Object >> iterator ()

Lista pública < saída <? >> saídas ()