Envolve um cálculo MLIR arbitrário expresso como um módulo com uma função main().
Esta operação não possui um kernel associado e não se destina a ser executada em uma sessão normal do TensorFlow. Em vez disso, ele deve ser usado para testes 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 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 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 uso:
{@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> } '''
Métodos Públicos
MlirPassthroughOp estático | |
Iterador< Operando <Objeto>> | iterador () |
Lista< Saída <?>> | saídas () |
Métodos herdados
Métodos Públicos
public static MlirPassthroughOp create (escopo do escopo , entradas Iterable< Operand <?>>, String mlirModule, List<Class<?>> Toutputs)
Método de fábrica para criar uma classe que envolve uma nova operação MlirPassthroughOp.
Parâmetros
escopo | escopo atual |
---|
Devoluções
- uma nova instância de MlirPassthroughOp