Searched refs:run_state (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 481 RunState run_state(step_id, &devices_); in RunInternal() local
482 run_state.rendez = new IntraProcessRendezvous(device_mgr_.get()); in RunInternal()
510 run_state.collective_executor.reset(new CollectiveExecutor::Handle( in RunInternal()
518 num_executors, run_state.rendez, [&run_state](const Status& ret) { in RunInternal()
520 mutex_lock l(run_state.mu_); in RunInternal()
521 run_state.status.Update(ret); in RunInternal()
523 run_state.executors_done.Notify(); in RunInternal()
529 args.rendezvous = run_state.rendez; in RunInternal()
531 (run_state.collective_executor ? run_state.collective_executor->get() in RunInternal()
537 args.tensor_store = &run_state.tensor_store; in RunInternal()
[all …]
|
| D | direct_session.h | 271 const ExecutorsAndKeys* executors_and_keys, RunState* run_state,
279 const ExecutorsAndKeys* executors_and_keys, const RunState* run_state);
287 void WaitForNotification(RunState* run_state, CancellationManager* cm,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 221 const RunState* run_state,
980 const RunStepRequestWrapper& req, const RunState* run_state, in CheckFetches() argument
984 for (const auto& input : run_state->pending_inputs) { in CheckFetches()
1532 RunState* run_state = in PartialRunSetup() local
1538 std::make_pair(handle, std::unique_ptr<RunState>(run_state))); in PartialRunSetup()
1625 RunState* run_state = nullptr; in DoPartialRun() local
1633 run_state = it->second.get(); in DoPartialRun()
1644 if (!run_state->step_started) { in DoPartialRun()
1645 run_state->step_started = true; in DoPartialRun()
1648 const auto count = run_state->count; in DoPartialRun()
[all …]
|