Обертывает произвольное вычисление MLIR, выраженное в виде модуля с функцией main().
Эта операция не имеет связанного ядра и не предназначена для выполнения в обычном сеансе TensorFlow. Вместо этого он предназначен для использования для тестирования или для особого случая, когда пользователь намеревается передать пользовательские вычисления MLIR через граф TensorFlow с намерением использовать специальные инструменты для его последующей обработки (при настройке на другую среду, например, на TensorFlow lite). Ожидается, что модуль MLIR будет иметь функцию main(), которая будет использоваться в качестве точки входа. Входные данные для операций будут переданы в качестве аргумента функции main(), а возвращаемые значения основной функции будут сопоставлены с выходными данными. Пример использования:
{@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> } '''
Публичные методы
статический MlirPassthroughOp | create ( область области действия, входы Iterable< Operand <?>>, String mlirModule, List<Class<?>> Toutputs) Фабричный метод для создания класса, обертывающего новую операцию MlirPassthroughOp. |
Итератор< Операнд <Объект>> | итератор () |
Список< Вывод <?>> | выходы () |
Унаследованные методы
Публичные методы
public static MlirPassthroughOp create ( область действия, входы Iterable< Operand <?>>, String mlirModule, List<Class<?>> Toutputs)
Фабричный метод для создания класса, обертывающего новую операцию MlirPassthroughOp.
Параметры
объем | текущий объем |
---|
Возврат
- новый экземпляр MlirPassthroughOp