MlirPassthroughOp

classe finale publique MlirPassthroughOp

Encapsule un calcul MLIR arbitraire exprimé sous forme de module avec une fonction main().

Cette opération n'a pas de noyau associé et n'est pas destinée à être exécutée dans une session TensorFlow standard. Au lieu de cela, il est destiné à être utilisé à des fins de test ou dans des cas particuliers dans lesquels un utilisateur a l'intention de transmettre un calcul MLIR personnalisé via un graphique TensorFlow dans le but d'avoir des outils personnalisés le traitant en aval (lorsqu'il cible un environnement différent, comme TensorFlow Lite par exemple). Le module MLIR devrait avoir une fonction main() qui sera utilisée comme point d'entrée. Les entrées des opérations seront passées en argument à la fonction main() et les valeurs renvoyées de la fonction principale mappées aux sorties. Exemple d'utilisation :

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>
 
 '''

Constantes

Chaîne

OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

MlirPassthroughOp statique	créer ( Scope scope, Iterable< Operand <?>> entrées, String mlirModule, List<Class<? extends TType >> Toutputs) Méthode d'usine pour créer une classe encapsulant une nouvelle opération MlirPassthroughOp.
Itérateur < Opérande < TType >>	itérateur ()
Liste< Sortie <?>>	les sorties ()

Méthodes héritées

De la classe org.tensorflow.op.RawOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .
Chaîne finale	àChaîne ()

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

Depuis l'interface org.tensorflow.op.Op

résuméExécutionEnvironnement	env () Renvoie l'environnement d'exécution dans lequel cette opération a été créée.
opération abstraite	op () Renvoie cette unité de calcul sous la forme d'une seule Operation .

Depuis l'interface java.lang.Iterable

vide abstrait	forEach (Consommateur <? super T> arg0)
Itérateur abstrait < Opérande < TType >>	itérateur ()
Spliterator abstrait < Opérande < TType >>	séparateur ()

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "MlirPassthroughOp"

Méthodes publiques

public static MlirPassthroughOp create ( Scope scope, Iterable< Operand <?>> entrées, String mlirModule, List<Class<? extends TType >> Toutputs)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération MlirPassthroughOp.

Paramètres

portée	portée actuelle

Retour

• une nouvelle instance de MlirPassthroughOp

Itérateur public < Opérande < TType >> itérateur ()

liste publique< Sortie <?>> sorties ()