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

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

Дочерние классы

class OpHintArgumentTracker

Методы

add_input

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

add_input(
    *args, **kwargs
)

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

add_inputs

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

add_inputs(
    *args, **kwargs
)

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

add_output

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

add_output(
    *args, **kwargs
)

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

add_outputs

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

add_outputs(
    *args, **kwargs
)

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