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 и жадное выполнение не включено, используется стандартный граф. |
log_dir | директория для записи файла журнала. |
op_log | (Необязательно) Протокол OpLogProto, который будет записан. Если не предоставлен, создаётся новый. |
run_meta | (Необязательно) Протокол RunMetadata, который помогает вычислению 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/api_docs/python/tf/compat/v1/profiler/write_op_log
