Searched refs:partial_run_mgr_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | partial_run_mgr_test.cc | 88 PartialRunMgr partial_run_mgr_; member in tensorflow::__anon3bb49c840111::StatusPropagationTest
112 partial_run_mgr_.FindOrCreate(step_id, &cancellation_manager); in TEST_P()
114 partial_run_mgr_.ExecutorDone(step_id, param.executor_status); in TEST_P()
115 partial_run_mgr_.PartialRunDone(step_id, in TEST_P()
128 partial_run_mgr_.FindOrCreate(step_id, &cancellation_manager); in TEST_P()
130 partial_run_mgr_.PartialRunDone(step_id, in TEST_P()
133 partial_run_mgr_.ExecutorDone(step_id, param.executor_status); in TEST_P()
|
| D | worker.cc | 297 bool is_new_partial_run = partial_run_mgr_.FindOrCreate(step_id, &cm); in DoPartialRunGraph()
317 partial_run_mgr_.ExecutorDone(step_id, s); in DoPartialRunGraph()
339 partial_run_mgr_.PartialRunDone(step_id, finish, s); in DoPartialRunGraph()
|
| D | worker.h | 119 PartialRunMgr partial_run_mgr_;
|