Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

MlirPassthroughOp

Organiza tus páginas con colecciones Guarda y categoriza el contenido según tus preferencias.
Clase public final 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 el 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. Ejemplo de uso:

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
crear ( Alcance alcance, Iterable < operando <? >> entradas, Cadena mlirModule, List <Clase <? >> Toutputs)
Método de fábrica para crear una clase que envuelva una nueva operación MlirPassthroughOp.
Iterador < operando <Object >>
Lista < salida <? >>
salidas ()

Métodos heredados

Métodos públicos

public static MlirPassthroughOp crear ( Alcance alcance, Iterable < operando <? >> entradas, Cadena mlirModule, List <Clase <? >> 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

pública Iterator < operando <Object >> iterador ()

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