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, чтобы позднее её можно было заменить альтернативной реализацией.

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

Аргументы
`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 "name" — метка; "tag" — тег для группировки нескольких аргументов, которые будут агрегированы. Например, строка 'cool_input'. По сути, в одну подсказку можно добавить несколько входов для параллельных операций, которые впоследствии будут объединены. Например, static_rnn создаёт несколько копий состояния или входов. "aggregate" — стратегия агрегации, допустимая только при значении tag, отличном от None. Допустимые значения: OpHint.AGGREGATE_FIRST, OpHint.AGGREGATE_LAST и OpHint.AGGREGATE_STACK. "index_override" — глобальный индекс, который следует использовать. Он соответствует порядку аргументов в создаваемой итоговой заглушке.

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

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

`add_inputs`

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

add_inputs(
    *args, **kwargs
)

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

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

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

`add_output`

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

add_output(
    *args, **kwargs
)

Добавляет обёрнутый выходной аргумент в подсказку.

Аргументы

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

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

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

`add_outputs`

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

add_outputs(
    *args, **kwargs
)

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

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

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

Переменные класса
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