Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py55 config.graph_options.rewrite_options.CopyFrom(
83 config.graph_options.rewrite_options.CopyFrom(
140 config.graph_options.rewrite_options.CopyFrom(
168 config.graph_options.rewrite_options.CopyFrom(
198 config.graph_options.rewrite_options.CopyFrom(
227 graph_options = config_pb2.GraphOptions(rewrite_options=rewrite_options)
228 return config_pb2.ConfigProto(graph_options=graph_options)
299 graph_options = config_pb2.GraphOptions(
301 session_config = config_pb2.ConfigProto(graph_options=graph_options)
316 config.graph_options.rewrite_options.CopyFrom(
Dtf_optimizer_test.py52 rewriter_config = config.graph_options.rewrite_options
80 rewriter_config = config.graph_options.rewrite_options
125 rewriter_config = config.graph_options.rewrite_options
Dcost_analyzer_tool.py85 config.graph_options.rewrite_options)
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_xla_test.py40 graph_options = config_pb2.GraphOptions(
44 graph_options=graph_options)
45 rewrite_options = cfg.graph_options.rewrite_options
Dcollective_ops_test.py189 rewrite_options = config.graph_options.rewrite_options
250 rewrite_options = config.graph_options.rewrite_options
526 graph_options = config_pb2.GraphOptions(
529 graph_options=graph_options)
530 rewrite_options = cfg.graph_options.rewrite_options
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(
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_grappler_test.py47 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
48 return config_pb2.ConfigProto(graph_options=graph_options)
Ddebug_graph_reconstruction_test.py49 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
50 return 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/core/distributed_runtime/
Dgraph_mgr.cc127 const GraphOptions& graph_options, const DebugOptions& debug_options, in InitItem() argument
142 graph_options.optimizer_options(), worker_env_->compute_pool, cluster_flr, in InitItem()
184 popts.scheduling_for_recvs = graph_options.enable_recv_scheduling(); in InitItem()
212 const auto& optimizer_opts = graph_options.optimizer_options(); in InitItem()
286 unit->build_cost_model = graph_options.build_cost_model(); in InitItem()
297 const GraphOptions& graph_options, const DebugOptions& debug_options, in Register() argument
301 Status s = InitItem(handle, gdef, session, graph_options, debug_options, in Register()
Dgraph_mgr.h78 WorkerSession* session, const GraphOptions& graph_options,
182 WorkerSession* session, 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/kernel_tests/
Dreduce_benchmark_test.py74 graph_options=config_pb2.GraphOptions(
90 graph_options=config_pb2.GraphOptions(
/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py52 graph_options = config_pb2.GraphOptions(rewrite_options=rewriter_config)
53 return config_pb2.ConfigProto(graph_options=graph_options)
444 cfg.graph_options.optimizer_options.opt_level = (
446 cfg.graph_options.optimizer_options.do_function_inlining = True
485 config.graph_options.optimizer_options.global_jit_level = global_jit_level
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc101 if (options.session_options->config.graph_options().place_pruned_graph() || in MakeForBaseGraph()
111 if (!options.session_options->config.graph_options().place_pruned_graph()) { in MakeForBaseGraph()
136 if (!(base_execution_state.session_options_->config.graph_options() in MakeForPrunedGraph()
138 options.session_options->config.graph_options().place_pruned_graph())) { in MakeForPrunedGraph()
272 if (!session_options_->config.graph_options().place_pruned_graph()) { in Extend()
648 if (session_options_->config.graph_options().place_pruned_graph()) { in OptimizeGraph()
875 !session_options_->config.graph_options().place_pruned_graph()) { in BuildGraph()
Dlower_functional_ops.cc104 options.session_options && options.session_options->config.graph_options() in Run()
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py45 config_proto.graph_options.rewrite_options.min_graph_nodes = -1
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py71 config.graph_options.rewrite_options.arithmetic_optimization = (
95 config.graph_options.rewrite_options.remapping = (
/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/util/
Dfunction_utils.py129 rewriter_config = config.graph_options.rewrite_options
/external/tensorflow/tensorflow/python/eager/
Dcontext.py947 config.graph_options.optimizer_options.global_jit_level = (
978 setattr(config.graph_options.rewrite_options,
988 setattr(config.graph_options.rewrite_options,
1009 config.graph_options.rewrite_options.min_graph_nodes = nodes
1032 rewrite_options = config.graph_options.rewrite_options
1556 level = self.config.graph_options.optimizer_options.global_jit_level
1572 rewrite_options = self.config.graph_options.rewrite_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)
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py63 graph_options=config_pb2.GraphOptions(
70 cfg.graph_options.rewrite_options.arithmetic_optimization = (
73 cfg.graph_options.rewrite_options.arithmetic_optimization = (
76 cfg.graph_options.rewrite_options.function_optimization = (
79 cfg.graph_options.rewrite_options.function_optimization = (
82 cfg.graph_options.rewrite_options.constant_folding = (
85 cfg.graph_options.rewrite_options.constant_folding = (
248 graph_options=config_pb2.GraphOptions(
346 graph_options=config_pb2.GraphOptions(
507 graph_options=config_pb2.GraphOptions(
[all …]
/external/tensorflow/tensorflow/python/client/
Dsession.py693 and config.graph_options.rewrite_options.auto_mixed_precision !=
697 new_config.graph_options.rewrite_options.auto_mixed_precision = (
700 elif (config.graph_options.rewrite_options.auto_mixed_precision !=
705 self._add_shapes = config.graph_options.infer_shapes
1756 config.graph_options.place_pruned_graph = True

1234