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 | |
Iterator < Operando <Object >> | iteração () |
List < Output <? >> | saídas () |
Métodos herdados
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