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
| Аргументы | |
|---|---|
graph | tf.Graph. Если значение None и нетерпеливое выполнение не включено, используется граф по умолчанию. |
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. |
| Возвращаемое значение | |
|---|---|
| Если cmd — 'scope' или 'graph', возвращает протокол GraphNodeProto. Если cmd — 'op' или 'code', возвращает протокол MultiGraphNodeProto. Побочный эффект: stdout/file/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/api_docs/python/tf/compat/v1/profiler/profile
