tf.compat.v1.lite.OpHint

Класс, помогающий создавать вызовы функций tflite. (устаревший)

tf.compat.v1.lite.OpHint(
    function_name, level=1, children_inputs_mappings=None, **kwargs
)

Он позволяет взять набор операций TensorFlow и аннотировать их создание таким образом, чтобы toco знал, как преобразовать его в tflite. Это встраивает псевдофункцию в граф TensorFlow. Это позволяет встраивать информацию об использовании API высокого уровня в реализацию TensorFlow низкого уровня, чтобы позже можно было заменить альтернативной реализацией.

По существу, любой «вход» в эту псевдооперацию подаётся в идентичность, и к этому входу добавляются атрибуты перед использованием им составляющих операций, из которых состоит псевдооперация. Аналогичный процесс выполняется с любым выходом, который необходимо экспортировать из текущей операции.

Аргументы
`function_name`	Название функции (имя пользовательской операции в tflite)
`level`	Уровень OpHint.
`children_inputs_mappings`	Карта сопоставления входных/выходных данных OpHint дочерних операций. children_inputs_mappings должно выглядеть так: "parent_first_child_input": [{"parent_input_index": num, "child_input_index": num}, ...] "parent_last_child_output": [{"parent_output_index": num, "child_output_index": num}, ...] "internal_children_input_output": [{"child_input_index": num, "child_output_index": num}, ...]
`**kwargs`	Аргументы ключевых слов любых постоянных атрибутов для функции.

Подклассы

class OpHintArgumentTracker

Методы

`add_input`

Просмотреть исходный код

add_input(
    *args, **kwargs
)

Добавить обернутый аргумент ввода в подсказку.

Аргументы

*args Входной тензор.

**kwargs Метка "имя" "тег" для группировки нескольких аргументов, которые будут агрегированы. Например, строка 'cool_input'. Таким образом, несколько входов могут быть добавлены к одной подсказке для параллельных операций, которые в конечном итоге будут объединены. Пример — static_rnn, который создаёт несколько копий состояния или входов. "aggregate" стратегия агрегации, которая допустима только для тега, отличного от None. Допустимые значения: OpHint.AGGREGATE_FIRST, OpHint.AGGREGATE_LAST и OpHint.AGGREGATE_STACK. "index_override" Глобальный индекс для использования. Он соответствует порядку аргументов в конечном фрагменте, который будет сгенерирован.

Аргументы
`*args`	Входной тензор.
`**kwargs`	Метка "имя" "тег" для группировки нескольких аргументов, которые будут агрегированы. Например, строка 'cool_input'. Таким образом, несколько входов могут быть добавлены к одной подсказке для параллельных операций, которые в конечном итоге будут объединены. Пример — static_rnn, который создаёт несколько копий состояния или входов. "aggregate" стратегия агрегации, которая допустима только для тега, отличного от None. Допустимые значения: OpHint.AGGREGATE_FIRST, OpHint.AGGREGATE_LAST и OpHint.AGGREGATE_STACK. "index_override" Глобальный индекс для использования. Он соответствует порядку аргументов в конечном фрагменте, который будет сгенерирован.

Возвращаемое значение
Обернутый входной тензор.

`add_inputs`

Просмотреть исходный код

add_inputs(
    *args, **kwargs
)

Добавить последовательность входов в вызов функции.

Аргументы
`*args`	Список входов для преобразования (должны быть Tf.Tensor).
`**kwargs`	Это позволяет использовать 'имена', которые должны быть списком имён.

Возвращаемое значение
Обернутые входы (заместители тождества с дополнительными метаданными). Это также тензоры tf.

`add_output`

Просмотреть исходный код

add_output(
    *args, **kwargs
)

Добавить обернутый аргумент вывода в подсказку.

Аргументы

*args Выходной тензор.

Аргументы
`*args`	Выходной тензор.
`**kwargs`	Метка "имя" "тег" для группировки нескольких аргументов, которые будут агрегированы. Например, строка 'cool_input'. Таким образом, несколько входов могут быть добавлены к одной подсказке для параллельных операций, которые в конечном итоге будут объединены. Пример — static_rnn, который создаёт несколько копий состояния или входов. "aggregate" стратегия агрегации, которая допустима только для тега, отличного от None. Допустимые значения: OpHint.AGGREGATE_FIRST, OpHint.AGGREGATE_LAST и OpHint.AGGREGATE_STACK. "index_override" Глобальный индекс для использования. Он соответствует порядку аргументов в конечном фрагменте, который будет сгенерирован.

Возвращаемое значение
Обернутый выходной тензор.

`add_outputs`

Просмотреть исходный код

add_outputs(
    *args, **kwargs
)

Добавить последовательность выходов в вызов функции.

Аргументы
`*args`	Список выходов для преобразования (должны быть tf.Tensor).
`**kwargs`	См.

Возвращаемое значение
Обернутые выходы (заместители тождества с дополнительными метаданными). Это также тензоры tf.

Переменные класса
AGGREGATE_FIRST	`'first'`
AGGREGATE_LAST	`'last'`
AGGREGATE_STACK	`'stack'`
CHILDREN_INPUTS_MAPPINGS	`'_tflite_children_ophint_inputs_mapping'`
FUNCTION_AGGREGATE_ATTR	`'_tflite_function_aggregate'`
FUNCTION_INPUT_INDEX_ATTR	`'_tflite_function_input_index'`
FUNCTION_LEVEL_ATTR	`'_tflite_ophint_level'`
FUNCTION_NAME_ATTR	`'_tflite_function_name'`
FUNCTION_OUTPUT_INDEX_ATTR	`'_tflite_function_output_index'`
FUNCTION_SORT_INDEX_ATTR	`'_tflite_function_sort_index'`
FUNCTION_UUID_ATTR	`'_tflite_function_uuid'`
TFLITE_INPUT_INDICES	`'_tflite_input_indices'`

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/lite/OpHint