Searched refs:disable_model_pruning (Results 1 – 18 of 18) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | custom_graph_optimizer_registry.h | 32 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
|
D | custom_graph_optimizer_registry.cc | 167 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()
|
D | meta_optimizer.cc | 210 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 …]
|
D | meta_optimizer_test.cc | 210 config_list.disable_model_pruning = true; in TEST_F()
|
/external/tensorflow/tensorflow/python/grappler/ |
D | memory_optimizer_test.py | 53 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/ |
D | grappler.cc | 114 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()
|
D | grappler.h | 102 TF_TriState disable_model_pruning; member
|
/external/tensorflow/tensorflow/python/debug/lib/ |
D | debug_grappler_test.py | 41 disable_model_pruning=False,
|
D | debug_gradients_test.py | 40 disable_model_pruning=True,
|
D | session_debug_testlib.py | 55 disable_model_pruning=True,
|
/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
D | utils.py | 40 rewriter_config.disable_model_pruning = True
|
/external/tensorflow/tensorflow/python/debug/wrappers/ |
D | framework_test.py | 145 disable_model_pruning=True)
|
D | local_cli_wrapper_test.py | 158 disable_model_pruning=True,
|
/external/tensorflow/tensorflow/python/debug/cli/ |
D | profile_analyzer_cli_test.py | 37 disable_model_pruning=True,
|
D | analyzer_cli_test.py | 75 disable_model_pruning=True,
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | rewriter_config.proto | 96 bool disable_model_pruning = 2; field
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | jit_test.py | 44 disable_model_pruning=True,
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | check_ops_test.py | 1022 config.graph_options.rewrite_options.disable_model_pruning = 1
|