tf.compat.v1.profiler.write_op_log
Записывает переданный журнал 'op_log' и добавляет дополнительную информацию о модели ниже.
tf.compat.v1.profiler.write_op_log(
graph, log_dir, op_log=None, run_meta=None, add_trace=True
)
API также присваивает операциям в tf.compat.v1.trainable_variables() тип операции '_trainable_variables'. Кроме того, API записывает статистику 'flops' для операций, для которых определена функция op.RegisterStatistics(). Вычисление flops зависит от форм тензоров, заданных в 'graph', которые могут быть неполными. Если указан 'run_meta', он по возможности дополняет информацию о формах.
| Аргументы | |
|---|---|
graph | tf.Graph. Если значение равно None и выполнение в eager-режиме не включено, используется граф по умолчанию. |
log_dir | Каталог для записи файла журнала. |
op_log | (Необязательно) Прототип OpLogProto, который нужно записать. Если он не указан, создаётся новый. |
run_meta | (Необязательно) Прототип RunMetadata, который помогает вычислять flops с использованием информации о формах во время выполнения. |
add_trace | Добавлять ли информацию о трассировке кода Python. Используется для поддержки представления "code". |
© 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/profiler/write_op_log
