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