tf.compat.v1.lite.experimental.convert_op_hints_to_stubs
Преобразует граф с подсказками LiteOp в операции-заглушки.
tf.compat.v1.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 | Определяемый граф, который необходимо преобразовать. |
write_callback | Указатель на функцию, которая может использоваться для записи промежуточных шагов преобразования графа (необязательно). |
| Возвращаемое значение | |
|---|---|
| Новый граф с заменой всех операций, содержащихся в 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/r2.3/api_docs/python/tf/compat/v1/lite/experimental/convert_op_hints_to_stubs
