diff options
| author |  Ayush Dubey <ayushd@google.com> | 2018-09-25 19:03:54 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 19:07:51 -0700 |
| commit | 8adf133448f822fcb866d69ea9d046512836834e (patch) | |
| tree | 08f39803d3975fe24e3497680009f18298715c18 /tensorflow/core/distributed_runtime | |
| parent | d84159b6694773a558c8eeef63ef79b4034b8fa7 (diff) | |
Set step_id in Executor Args to the step_id generated in MasterSession.
PiperOrigin-RevId: 214542049
Diffstat (limited to 'tensorflow/core/distributed_runtime')
| -rw-r--r-- | tensorflow/core/distributed_runtime/graph_mgr.cc | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/core/distributed_runtime/graph_mgr.cc b/tensorflow/core/distributed_runtime/graph_mgr.cc index f7a2967d00..3361819e43 100644 --- a/tensorflow/core/distributed_runtime/graph_mgr.cc +++ b/tensorflow/core/distributed_runtime/graph_mgr.cc @@ -475,10 +475,7 @@ void GraphMgr::StartParallelExecutors(const string& handle, int64 step_id, delete step_container; }); Executor::Args args; - { - mutex_lock l(mu_); - args.step_id = ++next_id_; - } + args.step_id = step_id; args.rendezvous = rendezvous; args.collective_executor = ce_handle ? ce_handle->get() : nullptr; args.cancellation_manager = cancellation_manager; |