tf.lite.experimental.convert_op_hints_to_stubs
Преобразует граф с подсказками LiteOp в операции-заглушки.
tf.lite.experimental.convert_op_hints_to_stubs(
session=None, graph_def=None, write_callback=(lambda graph_def, comments: None)
)
Это используется для подготовки к преобразованию toco сложных встроенных функций. Примечание: следует использовать только сеанс или graph_def, а не оба.
| Аргументы | |
|---|---|
session | Сеанс TensorFlow, содержащий граф для преобразования. |
graph_def | Граф def, который мы должны преобразовать. |
write_callback | Указатель на функцию, который можно использовать для записи промежуточных шагов преобразования графа (необязательно). |
| Возвращаемое значение | |
|---|---|
| Новый graphdef, в котором все операции, содержащиеся в OpHints, заменены вызовом одной операции с правильными параметрами. |
| Возможные исключения | |
|---|---|
ValueError | Если предоставлены как сеанс, так и graph_def. |
© 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/experimental/convert_op_hints_to_stubs