Home
last modified time | relevance | path

Searched refs:BuildGraphOptions (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.h117 const BuildGraphOptions& subgraph_options,
142 Status BuildGraph(const BuildGraphOptions& options,
174 Status InitBaseGraph(const BuildGraphOptions& options);
187 Status PruneGraph(const BuildGraphOptions& options, Graph* graph,
191 const BuildGraphOptions& options, std::unique_ptr<Graph>* optimized_graph,
Dstats_publisher_interface.h55 const string&, const BuildGraphOptions&, const SessionOptions&)>
59 const string& session, const BuildGraphOptions& bopts,
Dgraph_execution_state.cc97 TF_RETURN_IF_ERROR(ret->InitBaseGraph(BuildGraphOptions())); in MakeForBaseGraph()
106 const BuildGraphOptions& subgraph_options, in MakeForPrunedGraph()
218 TF_RETURN_IF_ERROR(new_execution_state->InitBaseGraph(BuildGraphOptions())); in Extend()
442 const BuildGraphOptions& options, Graph* graph, in PruneGraph()
547 Status GraphExecutionState::InitBaseGraph(const BuildGraphOptions& options) { in InitBaseGraph()
596 const BuildGraphOptions& options, std::unique_ptr<Graph>* optimized_graph, in OptimizeGraph()
736 Status GraphExecutionState::BuildGraph(const BuildGraphOptions& options, in BuildGraph()
791 if (collective_graph_key == BuildGraphOptions::kNoCollectiveGraphKey) { in BuildGraph()
Dbuild_graph_options.h27 struct BuildGraphOptions { struct
Ddirect_session.h157 int64 collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey;
211 int64 collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey;
239 const BuildGraphOptions& options,
Dstats_publisher_interface.cc45 const string& session, const BuildGraphOptions& bopts, in CreateNoOpStatsPublisher()
Dbuild_graph_options.cc22 string BuildGraphOptions::DebugString() const { in DebugString()
Ddirect_session.cc486 BuildGraphOptions::kNoCollectiveGraphKey) { in RunInternal()
488 BuildGraphOptions::kNoCollectiveGraphKey) { in RunInternal()
1190 BuildGraphOptions options; in CreateExecutors()
1458 const BuildGraphOptions& subgraph_options, in CreateGraphs()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc67 ReffedClientGraph(const string& handle, const BuildGraphOptions& bopts, in ReffedClientGraph()
110 const BuildGraphOptions& build_graph_options() { return bg_opts_; } in build_graph_options()
226 const BuildGraphOptions bg_opts_;
1085 BuildGraphOptions* opts) { in BuildBuildGraphOptions()
1112 BuildGraphOptions* opts) { in BuildBuildGraphOptions()
1127 uint64 HashBuildGraphOptions(const BuildGraphOptions& opts) { in HashBuildGraphOptions()
1150 string BuildGraphOptionsString(const BuildGraphOptions& opts) { in BuildGraphOptionsString()
1163 if (opts.collective_graph_key != BuildGraphOptions::kNoCollectiveGraphKey) { in BuildGraphOptionsString()
1431 Status MasterSession::StartStep(const BuildGraphOptions& opts, bool is_partial, in StartStep()
1479 if (graph_key == BuildGraphOptions::kNoCollectiveGraphKey) { in NewStepId()
[all …]
Dmaster_session.h226 Status StartStep(const BuildGraphOptions& opts, bool is_partial,
Dgraph_mgr.cc429 item->collective_graph_key != BuildGraphOptions::kNoCollectiveGraphKey in ExecuteAsync()