tf.compat.v1.lite.OpHint.OpHintArgumentTracker

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

tf.compat.v1.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`	Если индексы несогласованы.

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