tf.lite.OpHint

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

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.lite.OpHint

tf.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 Метка "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`	Это позволяет использовать 'names', которые должны быть списком имен.

Возвращаемое значение
Обернутые входы (заместители идентичности, которые имеют дополнительную метаданные). Это также тензоры 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/r1.15/api_docs/python/tf/lite/OpHint