Обратите внимание, что оптимизации применяются только в режиме графа (внутри tf.function). Кроме того, поскольку это экспериментальные параметры, список может быть изменён.
Аргументы
options
Словарь экспериментальных параметров оптимизатора для настройки. Допустимые ключи:
layout_optimizer: Оптимизировать макеты тензоров, например, это позволит использовать макет NCHW на GPU, что быстрее.
constant_folding: Сворачивание констант Статически определять значение тензоров, когда это возможно, и материализовать результат с помощью констант.
shape_optimization: Упрощение вычислений, выполняемых над формами.
remapping: Переназначение подграфов на более эффективные реализации.
arithmetic_optimization: Упрощение арифметических операций с помощью устранения общих подвыражений и упрощения арифметических операций.
dependency_optimization: Управление оптимизациями зависимостей. Удаление избыточных зависимостей, что может позволить другие оптимизации. Этот оптимизатор также необходим для обрезки узлов Identity и NoOp.
loop_optimization: Оптимизации циклов.
function_optimization: Оптимизации функций и встраивание.
debug_stripper: Удаляет из графа узлы, связанные с отладкой.
disable_model_pruning: Отключить удаление ненужных операций из графа
scoped_allocator_optimization: Попробовать выделить независимые выходные данные некоторых операций непрерывно, чтобы объединить или устранить последующие операции.
pin_to_host_optimization: Принудительно поместить небольшие операции на ЦП.
implementation_selector: Включить обмен реализациями ядер на основе размещения устройства.
auto_mixed_precision: Изменить определённые операции float32 на float16 на GPU Volta и выше. Без использования масштабирования потерь это может привести к численному подтечению (см. keras.mixed_precision.experimental.LossScaleOptimizer).
disable_meta_optimizer: Отключить весь мета-оптимизатор.
min_graph_nodes: Минимальное количество узлов в графе для оптимизации. Для более мелких графиков оптимизация пропускается.
