| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 56 int64 collective_graph_key) in ClientGraph()
61 collective_graph_key(collective_graph_key) {} in ClientGraph()
68 int64 collective_graph_key; member
|
| D | build_graph_options.cc | 35 if (collective_graph_key != kNoCollectiveGraphKey) { in DebugString()
36 strings::StrAppend(&rv, "\ncollective_graph_key: ", collective_graph_key); in DebugString()
|
| D | direct_session.cc | 485 if (executors_and_keys->collective_graph_key != in RunInternal()
487 if (run_options.experimental().collective_graph_key() != in RunInternal()
491 if (run_options.experimental().collective_graph_key() != in RunInternal()
492 executors_and_keys->collective_graph_key) { in RunInternal()
495 run_options.experimental().collective_graph_key(), in RunInternal()
497 executors_and_keys->collective_graph_key); in RunInternal()
748 run_state_args.collective_graph_key = in Run()
749 run_options.experimental().collective_graph_key(); in Run()
756 collective_graph_key_ = executors_and_keys->collective_graph_key; in Run()
1193 options.collective_graph_key = in CreateExecutors()
[all …]
|
| D | build_graph_options.h | 36 int64 collective_graph_key = kNoCollectiveGraphKey; member
|
| D | direct_session.h | 157 int64 collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey; member
211 int64 collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey; member
243 DataTypeVector* output_types, int64* collective_graph_key);
|
| D | graph_execution_state.cc | 790 int64 collective_graph_key = options.collective_graph_key; in BuildGraph() local
791 if (collective_graph_key == BuildGraphOptions::kNoCollectiveGraphKey) { in BuildGraph()
824 collective_graph_key = hash; in BuildGraph()
839 rewrite_metadata.fetch_types, collective_graph_key)); in BuildGraph()
|
| D | direct_session_test.cc | 2255 int64* collective_graph_key) { in RunGraphWithCollectiveFunctions() argument
2270 *collective_graph_key = direct_session->collective_graph_key_; in RunGraphWithCollectiveFunctions()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.h | 79 const DebugOptions& debug_options, int64 collective_graph_key,
143 int64 collective_graph_key; member
182 const DebugOptions& debug_options, int64 collective_graph_key,
|
| D | graph_mgr.cc | 125 int64 collective_graph_key, in InitItem() argument
129 item->collective_graph_key = collective_graph_key; in InitItem()
285 int64 collective_graph_key, in Register() argument
290 collective_graph_key, cluster_flr, item); in Register()
429 item->collective_graph_key != BuildGraphOptions::kNoCollectiveGraphKey in ExecuteAsync()
|
| D | master_session.cc | 82 client_graph_before_register_->collective_graph_key) { in ReffedClientGraph()
112 int64 collective_graph_key() { return collective_graph_key_; } in collective_graph_key() function in tensorflow::MasterSession::ReffedClientGraph
1102 opts->collective_graph_key = in BuildBuildGraphOptions()
1103 req.options().experimental().collective_graph_key(); in BuildBuildGraphOptions()
1163 if (opts.collective_graph_key != BuildGraphOptions::kNoCollectiveGraphKey) { in BuildGraphOptionsString()
1164 strings::StrAppend(&buf, "\nGK: ", opts.collective_graph_key); in BuildGraphOptionsString()
1636 if (req.options().experimental().collective_graph_key() != in DoPartialRun()
1819 if (rcg->collective_graph_key() != in PostRunCleanup()
1821 env_->collective_executor_mgr->RetireStepId(rcg->collective_graph_key(), in PostRunCleanup()
1881 uint64 step_id = NewStepId(rcg->collective_graph_key()); in DoRunWithLocalExecution()
[all …]
|
| D | master_session.h | 180 int64 collective_graph_key; member
|
| D | worker.cc | 77 request->collective_graph_key(), session->cluster_flr.get(), in RegisterGraphAsync()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 46 run_options.experimental.collective_graph_key = 1
106 run_options.experimental.collective_graph_key = 1
128 run_options.experimental.collective_graph_key = 1
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-options.-experimental.pbtxt | 6 name: "collective_graph_key"
|
| D | tensorflow.-run-options.pbtxt | 53 name: "collective_graph_key"
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 129 // collective_graph_key will coordinate to use the same step_id
132 int64 collective_graph_key = 7; field
|
| D | config.proto | 498 int64 collective_graph_key = 1; field
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | cross_device_ops_test.py | 519 run_options.experimental.collective_graph_key = 6
|