tf.profiler.write_op_log
Записывает предоставленный 'op_log' и добавляет дополнительную информацию о модели ниже.
tf.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. Используется для поддержки "code" просмотра. |
© 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/profiler/write_op_log