MlirPassthroughOp

publiczna klasa końcowa MlirPassthroughOp

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
utwórz (zakres zakresu , wejścia Iterable< Operand <?>>, String mlirModule, List<Class<?>> Tooutputs)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację MlirPassthroughOp.
Iterator< Operand <Obiekt>>
Lista< Wyjście <?>>

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

publiczny Iterator< Operand <Object>> iterator ()

publiczna Lista< Wyjście <?>> wyjścia ()