Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py55 config.graph_options.rewrite_options.CopyFrom(
82 config.graph_options.rewrite_options.CopyFrom(
139 config.graph_options.rewrite_options.CopyFrom(
167 config.graph_options.rewrite_options.CopyFrom(
197 config.graph_options.rewrite_options.CopyFrom(
226 graph_options = config_pb2.GraphOptions(rewrite_options=rewrite_options)
227 return config_pb2.ConfigProto(graph_options=graph_options)
298 graph_options = config_pb2.GraphOptions(
300 session_config = config_pb2.ConfigProto(graph_options=graph_options)
314 config.graph_options.rewrite_options.CopyFrom(
Dtf_optimizer_test.py51 rewriter_config = config.graph_options.rewrite_options
76 rewriter_config = config.graph_options.rewrite_options
121 rewriter_config = config.graph_options.rewrite_options
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc123 const GraphOptions& graph_options, in InitItem() argument
143 item->lib_def.get(), graph_options.optimizer_options(), in InitItem()
172 popts.scheduling_for_recvs = graph_options.enable_recv_scheduling(); in InitItem()
200 const auto& optimizer_opts = graph_options.optimizer_options(); in InitItem()
272 unit->build_cost_model = graph_options.build_cost_model(); in InitItem()
283 const GraphOptions& graph_options, in Register() argument
289 Status s = InitItem(session, gdef, graph_options, debug_options, in Register()
Dgraph_mgr.h78 const GraphOptions& graph_options,
181 const GraphOptions& graph_options,
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.cc51 return options_.config.graph_options().build_cost_model_after(); in NumWarmupSteps()
65 return options_.config.graph_options().build_cost_model() != 0; in DetailedStatsEnabled()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_graph_reconstruction_test.py50 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
51 return config_pb2.ConfigProto(graph_options=graph_options)
Dstepper_test.py62 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
63 config = config_pb2.ConfigProto(graph_options=graph_options)
608 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
609 config = config_pb2.ConfigProto(graph_options=graph_options)
742 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
743 config = config_pb2.ConfigProto(graph_options=graph_options)
Ddebug_gradients_test.py46 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
47 config = config_pb2.ConfigProto(graph_options=graph_options)
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduce_benchmark_test.py74 graph_options=config_pb2.GraphOptions(
90 graph_options=config_pb2.GraphOptions(
Dparameterized_truncated_normal_op_test.py249 config = config_pb2.ConfigProto(graph_options=config_pb2.GraphOptions(
286 graph_options=config_pb2.GraphOptions(
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc96 if (!ret->session_options_->config.graph_options().place_pruned_graph()) { in MakeForBaseGraph()
109 DCHECK(options.session_options->config.graph_options().place_pruned_graph()); in MakeForPrunedGraph()
214 if (!session_options_->config.graph_options().place_pruned_graph()) { in Extend()
554 session_options_->config.graph_options().place_pruned_graph()) { in InitBaseGraph()
599 if (session_options_->config.graph_options().place_pruned_graph()) { in OptimizeGraph()
764 !session_options_->config.graph_options().place_pruned_graph()) { in BuildGraph()
/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py51 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
52 return config_pb2.ConfigProto(graph_options=graph_options)
438 cfg.graph_options.optimizer_options.opt_level = (
440 cfg.graph_options.optimizer_options.do_function_inlining = True
479 config.graph_options.optimizer_options.global_jit_level = global_jit_level
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py233 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_cfg)
235 graph_options = config_pb2.GraphOptions()
237 graph_options.rewrite_options.CopyFrom(
241 gpu_options=self._GetGPUOptions(), graph_options=graph_options)
/external/tensorflow/tensorflow/python/ops/
Dconv2d_benchmark.py126 config.graph_options.optimizer_options.opt_level = -1
127 rewrite_options = config.graph_options.rewrite_options
Dsplit_benchmark.py92 config = config_pb2.ConfigProto(graph_options=config_pb2.GraphOptions(
Dconcat_benchmark.py101 config = config_pb2.ConfigProto(graph_options=config_pb2.GraphOptions(
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py42 config_proto.graph_options.rewrite_options.min_graph_nodes = -1
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework_test.py150 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
151 return config_pb2.ConfigProto(graph_options=graph_options)
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py43 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
44 return config_pb2.ConfigProto(graph_options=graph_options)
Dstepper_cli_test.py151 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
152 config = config_pb2.ConfigProto(graph_options=graph_options)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py183 grappler_session_config.graph_options.HasField("rewrite_options")):
184 rewriter_config = grappler_session_config.graph_options.rewrite_options
186 grappler_session_config.graph_options.rewrite_options.CopyFrom(
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py62 graph_options=config_pb2.GraphOptions(
69 cfg.graph_options.rewrite_options.arithmetic_optimization = (
72 cfg.graph_options.rewrite_options.arithmetic_optimization = (
75 cfg.graph_options.rewrite_options.function_optimization = (
78 cfg.graph_options.rewrite_options.function_optimization = (
81 cfg.graph_options.rewrite_options.constant_folding = (
84 cfg.graph_options.rewrite_options.constant_folding = (
247 graph_options=config_pb2.GraphOptions(
345 graph_options=config_pb2.GraphOptions(
504 graph_options=config_pb2.GraphOptions(
[all …]
/external/tensorflow/tensorflow/python/util/
Dfunction_utils.py126 rewriter_config = config.graph_options.rewrite_options
/external/tensorflow/tensorflow/examples/tutorials/mnist/
Dmnist_softmax_xla.py65 config.graph_options.optimizer_options.global_jit_level = jit_level
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmulti_worker_test_base.py276 config.graph_options.optimizer_options.opt_level = -1
277 config.graph_options.rewrite_options.constant_folding = (

123