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>>	iterator ()
Lista< Wyjście <?>>	wyjścia ()

Metody dziedziczone

Z klasy org.tensorflow.op.PrimitiveOp

końcowa wartość logiczna	równa się (obiekt obiektu)
końcowy wew	hashCode ()
Operacja	op () Zwraca podstawową Operation
ostatni ciąg	doString ()

Z klasy java.lang.Object

wartość logiczna	równa się (Obiekt arg0)
ostatnia klasa<?>	pobierzKlasę ()
wew	hashCode ()
ostateczna pustka	powiadomić ()
ostateczna pustka	powiadom wszystkich ()
Strunowy	doString ()
ostateczna pustka	czekaj (długi arg0, int arg1)
ostateczna pustka	czekaj (długi arg0)
ostateczna pustka	Czekać ()

Z interfejsu java.lang.Iterable

próżnia	forEach (Konsument<? super T> arg0)
abstrakcyjny Iterator< Operand <Obiekt>>	iterator ()
Spliterator< Operand <Obiekt>>	rozdzielacz ()

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

Zwroty

• nowa instancja MlirPassthroughOp

publiczny Iterator< Operand <Object>> iterator ()

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