tf.compat.v1.lite.OpHint

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

Он позволяет взять набор операций 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 Метка «name», метка «tag» для группировки нескольких аргументов, которые будут агрегированы. Например, строка «cool_input». В основном, несколько входов могут быть добавлены в одну подсказку для параллельных операций, которые в конечном итоге будут объединены. Примером является static_rnn, который создает несколько копий состояния или входных данных. «aggregate» стратегия агрегации, которая действительна только для метки не None. Допустимые значения OpHint.AGGREGATE_FIRST, OpHint.AGGREGATE_LAST и OpHint.AGGREGATE_STACK. «index_override» Глобальный индекс для использования. Это соответствует порядку аргументов в окончательной вызова функции, которая будет сгенерирована.

Аргументы
`*args`	Входной тензор.
`**kwargs`	Метка «name», метка «tag» для группировки нескольких аргументов, которые будут агрегированы. Например, строка «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`	Метка «name», метка «tag» для группировки нескольких аргументов, которые будут агрегированы. Например, строка «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'

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