tf.compat.v1.profiler.ProfileOptionBuilder

Построитель опций для профилирования API.

tf.compat.v1.profiler.ProfileOptionBuilder(
    options=None
)

Для ознакомления с опциями, см. https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler/g3doc/options.md

# Users can use pre-built options:
opts = (
    tf.profiler.ProfileOptionBuilder.trainable_variables_parameter())

# Or, build your own options:
opts = (tf.compat.v1.profiler.ProfileOptionBuilder()
    .with_max_depth(10)
    .with_min_micros(1000)
    .select(['accelerator_micros'])
    .with_stdout_output()
    .build()

# Or customize the pre-built options:
opts = (tf.compat.v1.profiler.ProfileOptionBuilder(
    tf.profiler.ProfileOptionBuilder.time_and_memory())
    .with_displaying_options(show_name_regexes=['.*rnn.*'])
    .build())

# Finally, profiling with the options:
_ = tf.compat.v1.profiler.profile(tf.compat.v1.get_default_graph(),
                        run_meta=run_meta,
                        cmd='scope',
                        options=opts)

Методы

account_displayed_op_only

Посмотреть исходный код

account_displayed_op_only(
    is_true
)

Учитывать только статистику узлов профилировщика, отображаемых в результатах.

build

Посмотреть исходный код

build()

Создать опции профилирования.

float_operation

Посмотреть исходный код

@staticmethod
float_operation()

Опции, используемые для профилирования операций с плавающей точкой.

См. https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler/g3doc/profile_model_architecture.md для замечаний о расчете операций с плавающей точкой.

order_by

Посмотреть исходный код

order_by(
    attribute
)

Сортировать отображаемые узлы профилировщика по атрибуту.

Поддерживаемые атрибуты включают микросекунды, байты, количество, параметры и т. д. https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler/g3doc/options.md

select

Посмотреть исходный код

select(
    attributes
)

Выбрать атрибуты для отображения.

См. https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler/g3doc/options.md для поддерживаемых атрибутов.

time_and_memory

Посмотреть исходный код

@staticmethod
time_and_memory(
    min_micros=1,
    min_bytes=1,
    min_accelerator_micros=0,
    min_cpu_micros=0,
    min_peak_bytes=0,
    min_residual_bytes=0,
    min_output_bytes=0
)

Показать время выполнения и потребление памяти операций.

trainable_variables_parameter

Посмотреть исходный код

@staticmethod
trainable_variables_parameter()

Опции, используемые для профилирования параметров обучаемых переменных.

Обычно используется вместе с представлением 'scope'.

with_accounted_types

Посмотреть исходный код

with_accounted_types(
    account_type_regexes
)

Селективное подсчет статистики на основе типов узлов.

Здесь «типы» — это свойства узлов профилировщика. Профилировщик по умолчанию рассматривает имя устройства (например, /job:xx/.../device:GPU:0) и тип операции (например, MatMul) как свойства узлов профилировщика. Пользователь также может ассоциировать настраиваемые «типы» с узлами профилировщика через протокол OpLogProto.

Например, пользователь может выбрать узлы профилировщика, размещенные на gpu:0, с помощью: account_type_regexes=['.*gpu:0.*']

Если ни одно из свойств узла не соответствует указанным выражениям регулярных выражений, узел не отображается и не учитывается.

with_empty_output

Посмотреть исходный код

with_empty_output()

Не генерировать побочные выходы.

with_file_output

Посмотреть исходный код

with_file_output(
    outfile
)

Вывести результат в файл.

with_max_depth

Просмотреть исходный код

with_max_depth(
    max_depth
)

Установите максимальную глубину отображения.

Глубина зависит от отображения профиля. Для просмотра «scope» это глубина иерархии именованных областей (дерева), для просмотра «op» это количество типов операций (список) и т. д.

with_min_execution_time

Просмотреть исходный код

with_min_execution_time(
    min_micros=0, min_accelerator_micros=0, min_cpu_micros=0
)

Отображать узлы профилировщика, потребляющие не менее 'min_micros'.

with_min_float_operations

Просмотреть исходный код

with_min_float_operations(
    min_float_ops
)

Отображать только узлы профилировщика, потребляющие не менее 'min_float_ops'.

См. https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/profiler/g3doc/profile_model_architecture.md о замечаниях по расчету операций с плавающей точкой.

with_min_memory

Просмотреть исходный код

with_min_memory(
    min_bytes=0, min_peak_bytes=0, min_residual_bytes=0, min_output_bytes=0
)

Отображать только узлы профилировщика, потребляющие не менее 'min_bytes'.

with_min_occurrence

Просмотреть исходный код

with_min_occurrence(
    min_occurrence
)

Отображать только узлы профилировщика, содержащие не менее 'min_occurrence' узлов графа.

«Узел» означает узел вывода профилировщика, который может быть строкой Python (вид кода), типом операции (вид op) или узлом графа (вид графа/области). Строка Python включает все узлы графа, созданные этой строкой, а тип операции включает все узлы графа этого типа.

with_min_parameters

Просмотреть исходный код

with_min_parameters(
    min_params
)

Отображать только узлы профилировщика, содержащие не менее 'min_params' параметров.

«Параметры» обычно относятся к весам переменных TensorFlow. Это отражает «ёмкость» моделей.

with_node_names

Просмотреть исходный код

with_node_names(
    start_name_regexes=None,
    show_name_regexes=None,
    hide_name_regexes=None,
    trim_name_regexes=None
)

Регулярные выражения, используемые для выбора узлов профилировщика для отображения.

После оценки 'with_accounted_types', 'with_node_names' оцениваются следующим образом:

Для структуры данных профиля профилировщик сначала находит узлы профилировщика, соответствующие 'start_name_regexes', и начинает отображать узлы профилировщика с этого момента. Затем, если узел соответствует 'show_name_regexes' и не соответствует 'hide_name_regexes', он отображается. Если узел соответствует 'trim_name_regexes', профилировщик прекращает дальнейший поиск по этому ответвлению.

with_pprof_output

Просмотреть исходный код

with_pprof_output(
    pprof_file
)

Создать файл профиля pprof в формате gzip.

with_stdout_output

Просмотреть исходный код

with_stdout_output()

Вывести результат в стандартный вывод.

with_step

Просмотреть исходный код

with_step(
    step
)

Шаг профилирования, который использовать.

Здесь «шаг» относится к шагу, определенному API Profiler.add_step().

with_timeline_output

Просмотреть исходный код

with_timeline_output(
    timeline_file
)

Создать файл timeline в формате json.