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:
{@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> } '''
Metody publiczne
statyczny MlirPrzejścieOp | |
Iterator< Operand <Obiekt>> | iterator () |
Lista< Wyjście <?>> | wyjścia () |
Metody dziedziczone
Metody publiczne
public static MlirPassthroughOp create (zakres zakresu , wejścia Iterable< Operand <?>>, String mlirModule, List<Class<?>> Toutputs)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację MlirPassthroughOp.
Parametry
zakres | aktualny zakres |
---|
Powroty
- nowa instancja MlirPassthroughOp