Home
last modified time | relevance | path

Searched refs:disable_model_pruning (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dcustom_graph_optimizer_registry.h32 ConfigList(bool disable_model_pruning, in ConfigList()
34 : disable_model_pruning(disable_model_pruning), in ConfigList()
38 return (disable_model_pruning == other.disable_model_pruning) &&
41 bool disable_model_pruning; // Don't remove unnecessary ops from the graph. member
Dcustom_graph_optimizer_registry.cc167 cur_plugin_configs.disable_model_pruning, "\n"); in PrintPluginConfigsIfConflict()
188 if (cur_plugin_configs.disable_model_pruning == true) in GetPluginConfigs()
189 ret_plugin_configs.disable_model_pruning = true; in GetPluginConfigs()
205 if (user_config.disable_model_pruning != plugin_config.disable_model_pruning) in IsConfigsConflict()
Dmeta_optimizer.cc210 if (optimizer == "pruning" && !plugin_configs.disable_model_pruning) in MakeNewOptimizer()
285 if (!cfg_.disable_model_pruning() && !plugin_configs.disable_model_pruning) { in InitializeOptimizers()
618 user_cfg.disable_model_pruning = cfg_.disable_model_pruning(); in PrintUserAndPluginConfigs()
662 if (optimizer_name == "pruning") user_cfg.disable_model_pruning = true; in PrintUserAndPluginConfigs()
699 if (plugin_cfg.disable_model_pruning == true) in PrintUserAndPluginConfigs()
700 final_cfg.disable_model_pruning = true; in PrintUserAndPluginConfigs()
712 user_cfg.disable_model_pruning, "\t\t", in PrintUserAndPluginConfigs()
713 plugin_cfg.disable_model_pruning, "\t\t", in PrintUserAndPluginConfigs()
714 final_cfg.disable_model_pruning, "\n"); in PrintUserAndPluginConfigs()
801 if (cfg_.disable_model_pruning()) { in OptimizeGraph()
[all …]
Dmeta_optimizer_test.cc210 config_list.disable_model_pruning = true; in TEST_F()
/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py53 disable_model_pruning=True,
81 disable_model_pruning=True,
138 disable_model_pruning=True,
166 disable_model_pruning=True,
196 disable_model_pruning=True,
221 disable_model_pruning=True,
/external/tensorflow/tensorflow/c/experimental/grappler/
Dgrappler.cc114 if (tp_configs.disable_model_pruning == TF_TriState_On) in CGraphOptimizerRegister()
115 configs.disable_model_pruning = true; in CGraphOptimizerRegister()
117 configs.disable_model_pruning = false; in CGraphOptimizerRegister()
Dgrappler.h102 TF_TriState disable_model_pruning; member
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_grappler_test.py41 disable_model_pruning=False,
Ddebug_gradients_test.py40 disable_model_pruning=True,
Dsession_debug_testlib.py55 disable_model_pruning=True,
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dutils.py40 rewriter_config.disable_model_pruning = True
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework_test.py145 disable_model_pruning=True)
Dlocal_cli_wrapper_test.py158 disable_model_pruning=True,
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py37 disable_model_pruning=True,
Danalyzer_cli_test.py75 disable_model_pruning=True,
/external/tensorflow/tensorflow/core/protobuf/
Drewriter_config.proto96 bool disable_model_pruning = 2; field
/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py44 disable_model_pruning=True,
/external/tensorflow/tensorflow/python/kernel_tests/
Dcheck_ops_test.py1022 config.graph_options.rewrite_options.disable_model_pruning = 1