tf.lite.OpHint.OpHintArgumentTracker

Концептуально отслеживает индексы аргументов «функций OpHint».

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

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

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

tf.compat.v1.lite.OpHint.OpHintArgumentTracker

tf.lite.OpHint.OpHintArgumentTracker(
    function_name, unique_function_id, node_name_prefix, attr_name, level=1,
    children_inputs_mappings=None
)

Входы и аргументы этих функций оба используют экземпляр класса, чтобы они могли иметь независимые нумерации.

Аргументы
`function_name`	Название функции, для которой отслеживаются аргументы.
`unique_function_id`	UUID функции, для которой отслеживаются аргументы.
`node_name_prefix`	Способ именования создаваемых идентификаторов.
`attr_name`	Имя атрибута для хранения индекса этого подсказки. Например, FUNCTION_INPUT_INDEX или FUNCTION_OUTPUT_INDEX
`level`	Иерархический уровень узла Ophint, число.
`children_inputs_mappings`	Сопоставление входов/выходов для подсказок дочерних элементов.

Методы

`add`

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

add(
    arg, tag=None, name=None, aggregate=None, index_override=None
)

Возвращает обернутый тензор входного тензора как аргумент.

Аргументы
`arg`	Тензор TensorFlow, который должен рассматриваться как аргумент.
`tag`	Строковый тег для идентификации аргументов, которые должны быть упакованы.
`name`	Имя аргумента. Оно включается в имена операций подсказок Identity.
`aggregate`	Стратегия агрегирования. Допустимые значения — OpHint.AGGREGATE_FIRST, OpHint.AGGREGATE_LAST и OpHint.AGGREGATE_STACK. Обратите внимание, что агрегирование допустимо только при указании тега.
`index_override`	Укажите, какой индекс вход/выход должен быть у этого элемента в конечном фрагменте. Например, add(arg0, index=1); add(arg1, index=0) приведет к тому, что конечный фрагмент будет таким: stub_func(inputs[arg1, arg0], outputs=[]), а не к порядку вызова по умолчанию.

Возвращаемое значение
Тензор, представляющий обернутый аргумент.

Исключения
`ValueError`	При несогласованности индексов.

© 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/OpHintArgumentTracker