| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 57 int64 collective_graph_key) in ClientGraph()
62 collective_graph_key(collective_graph_key) {} in ClientGraph()
69 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 | build_graph_options.h | 36 int64 collective_graph_key = kNoCollectiveGraphKey; member
|
| D | direct_session.cc | 535 if (executors_and_keys->collective_graph_key != in RunInternal()
537 if (run_options.experimental().collective_graph_key() != in RunInternal()
541 if (run_options.experimental().collective_graph_key() != in RunInternal()
542 executors_and_keys->collective_graph_key) { in RunInternal()
545 run_options.experimental().collective_graph_key(), in RunInternal()
547 executors_and_keys->collective_graph_key); in RunInternal()
853 run_state_args.collective_graph_key = in Run()
854 run_options.experimental().collective_graph_key(); in Run()
861 collective_graph_key_ = executors_and_keys->collective_graph_key; in Run()
1299 options.collective_graph_key = in CreateExecutors()
[all …]
|
| D | direct_session.h | 178 int64 collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey; member
238 int64 collective_graph_key = BuildGraphOptions::kNoCollectiveGraphKey; member
264 DataTypeVector* output_types, int64* collective_graph_key);
|
| D | graph_execution_state.cc | 901 int64 collective_graph_key = options.collective_graph_key; in BuildGraph() local
902 if (collective_graph_key == BuildGraphOptions::kNoCollectiveGraphKey) { in BuildGraph()
941 collective_graph_key = hash; in BuildGraph()
943 collective_graph_key = 0x8774aa605c729c72ULL; in BuildGraph()
958 rewrite_metadata.fetch_types, collective_graph_key)); in BuildGraph()
|
| D | direct_session_test.cc | 2714 int64* collective_graph_key) { in RunGraphWithCollectiveFunctions() argument
2729 *collective_graph_key = direct_session->collective_graph_key_; in RunGraphWithCollectiveFunctions()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.h | 80 const ConfigProto& config_proto, int64 collective_graph_key,
144 int64 collective_graph_key; member
184 const ConfigProto& config_proto, int64 collective_graph_key,
|
| D | graph_mgr.cc | 128 const ConfigProto& config_proto, int64 collective_graph_key, in InitItem() argument
131 item->collective_graph_key = collective_graph_key; in InitItem()
298 const ConfigProto& config_proto, int64 collective_graph_key, in Register() argument
302 config_proto, collective_graph_key, cluster_flr, item); in Register()
461 item->collective_graph_key != BuildGraphOptions::kNoCollectiveGraphKey in ExecuteAsync()
|
| D | master_session.cc | 86 client_graph_before_register_->collective_graph_key) { in ReffedClientGraph()
116 int64 collective_graph_key() { return collective_graph_key_; } in collective_graph_key() function in tensorflow::MasterSession::ReffedClientGraph
1159 opts->collective_graph_key = in BuildBuildGraphOptions()
1160 req.options().experimental().collective_graph_key(); in BuildBuildGraphOptions()
1220 if (opts.collective_graph_key != BuildGraphOptions::kNoCollectiveGraphKey) { in BuildGraphOptionsString()
1221 strings::StrAppend(&buf, "\nGK: ", opts.collective_graph_key); in BuildGraphOptionsString()
1728 if (req.options().experimental().collective_graph_key() != in DoPartialRun()
1911 if (rcg->collective_graph_key() != in PostRunCleanup()
1913 env_->collective_executor_mgr->RetireStepId(rcg->collective_graph_key(), in PostRunCleanup()
1973 uint64 step_id = NewStepId(rcg->collective_graph_key()); in DoRunWithLocalExecution()
[all …]
|
| D | master_session.h | 178 int64 collective_graph_key; member
|
| D | worker.cc | 84 request->config_proto(), request->collective_graph_key(), in RegisterGraphAsync()
|
| /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/python/ops/ |
| D | collective_ops_test.py | 86 run_options.experimental.collective_graph_key = 1
317 run_options.experimental.collective_graph_key = 1
345 run_options.experimental.collective_graph_key = 1
388 run_options.experimental.collective_graph_key = 1
410 run_options.experimental.collective_graph_key = 1
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 524 run_options.experimental.collective_graph_key = 5
705 run_options.experimental.collective_graph_key = 7
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 142 // collective_graph_key will coordinate to use the same step_id
145 int64 collective_graph_key = 7; field
|
| D | config.proto | 685 int64 collective_graph_key = 1; field
|