Zawija dowolne obliczenia MLIR wyrażone jako moduł z funkcją main().
Ta operacja nie ma skojarzonego jądra i nie jest przeznaczona do wykonywania w zwykłej sesji TensorFlow. Zamiast tego jest przeznaczony do testowania lub w specjalnych przypadkach, gdy użytkownik zamierza przekazać niestandardowe obliczenia MLIR przez wykres TensorFlow z zamiarem przetworzenia ich przez niestandardowe narzędzia w dalszej części procesu (w przypadku kierowania na inne środowisko, na przykład TensorFlow lite). Oczekuje się, że moduł MLIR będzie miał funkcję main(), która będzie używana jako punkt wejścia. Dane wejściowe operacji zostaną przekazane jako argument do funkcji main(), a zwrócone wartości funkcji main zostaną odwzorowane na dane wyjściowe. Przykładowe użycie:
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>
'''
Stałe
Smyczkowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
statyczny MlirPrzejścieOp | |
Iterator< Operand < TType >> | iterator () |
Lista< Wyjście <?>> | wyjścia () |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
public static MlirPassthroughOp create (zakres zakresu , wejścia Iterable< Operand <?>>, String mlirModule, List<Class<? Extends TType >> Toutputs)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację MlirPassthroughOp.
Parametry
zakres | aktualny zakres |
---|
Powroty
- nowa instancja MlirPassthroughOp