Searched refs:RetireStepId (Results 1 – 7 of 7) sorted by relevance
103 cme_->RetireStepId(6, x); in TEST_F()106 cme_->RetireStepId(7, x); in TEST_F()
57 void RetireStepId(int64 graph_key, int64 step_id) override;
158 void RpcCollectiveExecutorMgr::RetireStepId(int64 graph_key, int64 step_id) { in RetireStepId() function in tensorflow::RpcCollectiveExecutorMgr
1913 env_->collective_executor_mgr->RetireStepId(rcg->collective_graph_key(), in PostRunCleanup()
64 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() function
123 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() function
230 virtual void RetireStepId(int64 graph_key, int64 step_id) = 0;