Home
last modified time | relevance | path

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

12

/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.cc142 auto& rewriter_config = in TEST_F() local
144 rewriter_config.add_optimizers("TestOptimizer"); in TEST_F()
145 rewriter_config.set_min_graph_nodes(-1); in TEST_F()
161 auto& rewriter_config = in TEST_F() local
163 rewriter_config.add_optimizers("TestOptimizerWithParams"); in TEST_F()
164 auto* custom_config = rewriter_config.add_custom_optimizers(); in TEST_F()
183 auto& rewriter_config = in TEST_F() local
185 rewriter_config.add_optimizers("TestOptimizer"); in TEST_F()
186 auto customGraphOptimizer = rewriter_config.add_custom_optimizers(); in TEST_F()
188 rewriter_config.set_min_graph_nodes(-1); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drebatch_dataset_op.cc74 RewriterConfig rewriter_config; in CreateGrapplerRewriteConfig() local
75 rewriter_config.set_fail_on_optimizer_errors(true); in CreateGrapplerRewriteConfig()
76 rewriter_config.add_optimizers(kOptimizerName); in CreateGrapplerRewriteConfig()
77 rewriter_config.set_meta_optimizer_iterations( in CreateGrapplerRewriteConfig()
79 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateGrapplerRewriteConfig()
85 return rewriter_config; in CreateGrapplerRewriteConfig()
Dauto_shard_dataset_op.cc85 RewriterConfig rewriter_config; in CreateGrapplerRewriteConfig() local
86 rewriter_config.set_fail_on_optimizer_errors(true); in CreateGrapplerRewriteConfig()
87 rewriter_config.add_optimizers(kOptimizerName); in CreateGrapplerRewriteConfig()
88 rewriter_config.set_meta_optimizer_iterations( in CreateGrapplerRewriteConfig()
90 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateGrapplerRewriteConfig()
101 return rewriter_config; in CreateGrapplerRewriteConfig()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py84 rewriter_config = rewriter_config_pb2.RewriterConfig()
88 rewriter_config.layout_optimizer = off
89 rewriter_config.constant_folding = off
90 rewriter_config.shape_optimization = off
91 rewriter_config.remapping = off
92 rewriter_config.arithmetic_optimization = off
93 rewriter_config.dependency_optimization = off
94 rewriter_config.loop_optimization = off
95 rewriter_config.function_optimization = off
96 rewriter_config.debug_stripper = off
[all …]
Dint32_test.py68 rewriter_config=trt_test.OptimizerDisabledRewriterConfig())
Ddynamic_input_shapes_test.py88 rewriter_config=trt_test.OptimizerDisabledRewriterConfig())
Dbiasadd_matmul_test.py127 rewriter_config=trt_test.OptimizerDisabledRewriterConfig())
/external/tensorflow/tensorflow/core/kernels/data/
Doptimize_dataset_op.cc72 RewriterConfig rewriter_config; in CreateGrapplerRewriteConfig() local
73 rewriter_config.add_optimizers(kOptimizerName); in CreateGrapplerRewriteConfig()
74 rewriter_config.set_meta_optimizer_iterations( in CreateGrapplerRewriteConfig()
76 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateGrapplerRewriteConfig()
84 return rewriter_config; in CreateGrapplerRewriteConfig()
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_test.py51 rewriter_config = config.graph_options.rewrite_options
52 rewriter_config.optimizers.append('constfold')
53 rewriter_config.min_graph_nodes = -1
76 rewriter_config = config.graph_options.rewrite_options
77 rewriter_config.min_graph_nodes = -1
121 rewriter_config = config.graph_options.rewrite_options
122 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/python/util/
Dfunction_utils.py126 rewriter_config = config.graph_options.rewrite_options
127 rewriter_config.disable_meta_optimizer = True
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_graph_reconstruction_test.py46 rewriter_config = rewriter_config_pb2.RewriterConfig(
50 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
Dstepper_test.py58 rewriter_config = rewriter_config_pb2.RewriterConfig(
62 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
604 rewriter_config = rewriter_config_pb2.RewriterConfig(
608 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
738 rewriter_config = rewriter_config_pb2.RewriterConfig(
742 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)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py181 rewriter_config = None
184 rewriter_config = grappler_session_config.graph_options.rewrite_options
185 custom_rewriter_config = self.get_rewriter_config(rewriter_config)
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_pb_text_files.txt29 tensorflow/core/protobuf/rewriter_config.pb_text.cc
Dtf_proto_files.txt43 tensorflow/core/protobuf/rewriter_config.proto
Dproto_text_pb_cc_files.txt37 tensorflow/core/protobuf/rewriter_config.pb.cc
Dproto_text_pb_h_files.txt37 tensorflow/core/protobuf/rewriter_config.pb.h
/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.py156 rewriter_config = rewriter_config_pb2.RewriterConfig(
160 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)
Dstepper_cli_test.py149 rewriter_config = rewriter_config_pb2.RewriterConfig(
151 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py46 rewriter_config = rewriter_config_pb2.RewriterConfig(
51 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)

12