Home
last modified time | relevance | path

Searched refs:rewriter_config (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dutils.py24 def disable_non_trt_optimizers_in_rewriter_config(rewriter_config): argument
28 rewriter_config.arithmetic_optimization = off
29 rewriter_config.auto_mixed_precision = off
30 rewriter_config.auto_parallel.enable = False
31 rewriter_config.constant_folding = off
32 rewriter_config.debug_stripper = off
33 rewriter_config.dependency_optimization = off
35 rewriter_config.disable_meta_optimizer = False
36 rewriter_config.disable_model_pruning = True
37 rewriter_config.function_optimization = off
[all …]
Dtrt_convert_test.py81 def _GetConfigProto(self, rewriter_config=None): argument
85 if rewriter_config:
86 config.graph_options.rewrite_options.CopyFrom(rewriter_config)
356 rewriter_config=trt_convert.get_tensorrt_rewriter_config(
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.cc74 auto rewriter_config = in DisableOptimizer() local
76 rewriter_config->set_layout_optimizer(RewriterConfig::OFF); in DisableOptimizer()
77 rewriter_config->set_disable_model_pruning(true); in DisableOptimizer()
78 rewriter_config->set_function_optimization(RewriterConfig::OFF); in DisableOptimizer()
79 rewriter_config->set_arithmetic_optimization(RewriterConfig::OFF); in DisableOptimizer()
80 rewriter_config->set_loop_optimization(RewriterConfig::OFF); in DisableOptimizer()
81 rewriter_config->set_dependency_optimization(RewriterConfig::OFF); in DisableOptimizer()
82 rewriter_config->set_constant_folding(RewriterConfig::OFF); in DisableOptimizer()
83 rewriter_config->set_memory_optimization(RewriterConfig::NO_MEM_OPT); in DisableOptimizer()
84 rewriter_config->set_shape_optimization(RewriterConfig::OFF); in DisableOptimizer()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc147 auto& rewriter_config = in TEST_F() local
149 rewriter_config.add_optimizers("TestOptimizer"); in TEST_F()
150 rewriter_config.set_min_graph_nodes(-1); in TEST_F()
166 auto& rewriter_config = in TEST_F() local
168 rewriter_config.add_optimizers("TestOptimizerWithParams"); in TEST_F()
169 auto* custom_config = rewriter_config.add_custom_optimizers(); in TEST_F()
188 auto& rewriter_config = in TEST_F() local
190 rewriter_config.add_optimizers("TestOptimizer"); in TEST_F()
191 auto customGraphOptimizer = rewriter_config.add_custom_optimizers(); in TEST_F()
193 rewriter_config.set_min_graph_nodes(-1); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op.cc76 RewriterConfig rewriter_config; in CreateConfig() local
77 rewriter_config.set_fail_on_optimizer_errors(true); in CreateConfig()
78 rewriter_config.set_meta_optimizer_iterations(RewriterConfig::ONE); in CreateConfig()
80 rewriter_config.add_optimizers(kOptimizerName); in CreateConfig()
81 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateConfig()
96 return rewriter_config; in CreateConfig()
/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
Dbatch_op_rewriter_test.cc70 RewriterConfig_CustomGraphOptimizer rewriter_config; in MakeConfig() local
71 (*rewriter_config.mutable_parameter_map())["batch_op_rewrite_config"].set_s( in MakeConfig()
73 return rewriter_config; in MakeConfig()
86 RewriterConfig_CustomGraphOptimizer rewriter_config = MakeConfig(config); in TEST_P() local
88 TF_ASSERT_OK(optimizer.Init(&rewriter_config)); in TEST_P()
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_test.py52 rewriter_config = config.graph_options.rewrite_options
53 rewriter_config.optimizers.append('constfold')
54 rewriter_config.min_graph_nodes = -1
80 rewriter_config = config.graph_options.rewrite_options
81 rewriter_config.min_graph_nodes = -1
125 rewriter_config = config.graph_options.rewrite_options
126 rewriter_config.min_graph_nodes = -1
Dcost_analyzer_tool.py83 if FLAGS.rewriter_config is not None:
84 text_format.Merge(FLAGS.rewriter_config,
/external/tensorflow/tensorflow/core/kernels/data/
Doptimize_dataset_op.cc153 RewriterConfig rewriter_config; in CreateConfig() local
154 rewriter_config.add_optimizers(kOptimizerName); in CreateConfig()
155 rewriter_config.set_meta_optimizer_iterations(RewriterConfig::ONE); in CreateConfig()
156 rewriter_config.set_fail_on_optimizer_errors(true); in CreateConfig()
157 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateConfig()
170 return rewriter_config; in CreateConfig()
/external/tensorflow/tensorflow/python/util/
Dfunction_utils.py129 rewriter_config = config.graph_options.rewrite_options
130 rewriter_config.disable_meta_optimizer = True
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_grappler_test.py44 rewriter_config = rewriter_config_pb2.RewriterConfig(
47 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
Ddebug_graph_reconstruction_test.py45 rewriter_config = rewriter_config_pb2.RewriterConfig(
49 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
Ddebug_gradients_test.py43 rewriter_config = rewriter_config_pb2.RewriterConfig(
46 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
Dsession_debug_testlib.py59 rewriter_config = rewriter_config_pb2.RewriterConfig(
63 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework_test.py148 rewriter_config = rewriter_config_pb2.RewriterConfig(
150 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
Dlocal_cli_wrapper_test.py161 rewriter_config = rewriter_config_pb2.RewriterConfig(
165 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py40 rewriter_config = rewriter_config_pb2.RewriterConfig(
43 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
Danalyzer_cli_test.py63 rewriter_config = rewriter_config_pb2.RewriterConfig(
70 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
/external/tensorflow/tensorflow/core/protobuf/
DBUILD33 "rewriter_config.proto",
Dconfig.proto10 import "tensorflow/core/protobuf/rewriter_config.proto";
/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py47 rewriter_config = rewriter_config_pb2.RewriterConfig(
52 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
/external/tensorflow/tensorflow/python/profiler/
Dmodel_analyzer_test.py54 rewriter_config = rewriter_config_pb2.RewriterConfig(
56 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py503 rewriter_config = rewriter_config_pb2.RewriterConfig(
513 rewrite_options=rewriter_config))
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc277 RewriterConfig* rewriter_config = graph_opts->mutable_rewrite_options(); in CallOptPassesAndPlace() local
278 rewriter_config->set_disable_meta_optimizer(true); in CallOptPassesAndPlace()
/external/tensorflow/tensorflow/core/
DBUILD175 "//tensorflow/core/protobuf:rewriter_config.proto",