Conozca lo último en aprendizaje automático, IA generativa y más en el
Simposio WiML 2023.
MlirPassthroughOp
Organiza tus páginas con colecciones
Guarda y categoriza el contenido según tus preferencias.
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 normal de TensorFlow. En cambio, 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. Ejemplo 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 | crear ( ámbito de alcance, iterable< operando <?>> entradas, cadena mlirModule, lista<clase<?>> salidas) Método de fábrica para crear una clase que envuelve una nueva operación MlirPassthroughOp. |
Iterador< Operando <Objeto>> | |
Lista< Salida <?>> | |
Métodos Heredados
De la clase java.lang.Object booleano | es igual a (Objeto arg0) |
clase final<?> | obtenerClase () |
En t | código hash () |
vacío final | notificar () |
vacío final | notificar a todos () |
Cuerda | a la cadena () |
vacío final | espera (largo arg0, int arg1) |
vacío final | esperar (largo arg0) |
vacío final | espera () |
Desde la interfaz java.lang.Iterable vacío | forEach (Consumidor<? super T> arg0) |
Iterador abstracto< Operando <Objeto>> | iterador () |
Spliterator< Operando <Objeto>> | divisor () |
Métodos públicos
Método de fábrica para crear una clase que envuelve una nueva operación MlirPassthroughOp.
Devoluciones
- una nueva instancia de MlirPassthroughOp
public Iterator< Operando <Objeto>> iterador ()
public List< salida <?>> salidas ()
Salvo que se indique lo contrario, el contenido de esta página está sujeto a la licencia Atribución 4.0 de Creative Commons, y los ejemplos de código están sujetos a la licencia Apache 2.0. Para obtener más información, consulta las políticas del sitio de Google Developers. Java es una marca registrada de Oracle o sus afiliados. Cierto contenido está sujeto a la licencia de NumPy.
Última actualización: 2023-10-06 (UTC)
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Falta la información que necesito"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Muy complicado o demasiados pasos"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Desactualizado"
},{
"type": "thumb-down",
"id": "translationIssue",
"label":"Problema de traducción"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Problema con las muestras o los códigos"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Otro"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Fácil de comprender"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Resolvió mi problema"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Otro"
}]
{
"lastModified": "Última actualización: 2023-10-06 (UTC)",
"confidential": False
}