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', если предоставлен, дополняет информацию о форме с максимальной точностью.
| Args | |
|---|---|
graph | tf.Graph. Если None и eager execution не включен, используйте граф по умолчанию. |
log_dir | директория для записи файла журнала. |
op_log | (Необязательно) OpLogProto proto для записи. Если не указано, создается новый. |
run_meta | (Необязательно) RunMetadata proto, который помогает вычислению flops с использованием информации о форме времени выполнения. |
add_trace | Необходимо ли добавлять информацию о трассировке кода Python. Используется для поддержки представления "код". |
© 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/versions/r2.9/api_docs/python/tf/compat/v1/profiler/write_op_log
