tf.compat.v1.profiler.profile
Профилирование модели.
tf.compat.v1.profiler.profile(
graph=None,
run_meta=None,
op_log=None,
cmd='scope',
options=_DEFAULT_PROFILE_OPTIONS
)
Учебные пособия и примеры можно найти по адресу: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/profiler/g3doc/python_api.md
| Args | |
|---|---|
graph | tf.Graph. Если None и не включено Eager-выполнение, используется граф по умолчанию. |
run_meta | необязательный протокол tensorflow.RunMetadata. Необходим для поддержки профилирования информации времени выполнения, такой как время и память. |
op_log | протокол tensorflow.tfprof.OpLogProto. Пользователь может назначать «типы» узлам графа с помощью op_log. «Типы» позволяют пользователю гибко группировать и учитывать профили с помощью options['accounted_type_regexes']. |
cmd | строка. Либо 'op', 'scope', 'graph' или 'code'. Представление 'op' организует профиль с использованием типа операции. (например, MatMul) Представление 'scope' организует профиль с использованием области имен узла графа. Представление 'graph' организует профиль с использованием входов/выходов узла графа. Представление 'code' организует профиль с использованием стека вызовов Python. |
options | Словарь параметров. См. core/profiler/g3doc/options.md. |
| Returns | |
|---|---|
| Если cmd — 'scope' или 'graph', возвращает протокол GraphNodeProto. Если cmd — 'op' или 'code', возвращает протокол MultiGraphNodeProto. Побочный эффект: stdout/файл/timeline.json в зависимости от options['output'] |
© 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/profile