Searched refs:RetireStepId (Results 1 – 7 of 7) sorted by relevance
100 cme_->RetireStepId(6, x); in TEST_F()103 cme_->RetireStepId(7, x); in TEST_F()
56 void RetireStepId(int64 graph_key, int64 step_id) override;
155 void RpcCollectiveExecutorMgr::RetireStepId(int64 graph_key, int64 step_id) { in RetireStepId() function in tensorflow::RpcCollectiveExecutorMgr
1821 env_->collective_executor_mgr->RetireStepId(rcg->collective_graph_key(), in PostRunCleanup()
56 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() function
110 void RetireStepId(int64 graph_key, int64 step_id) override {} in RetireStepId() function
205 virtual void RetireStepId(int64 graph_key, int64 step_id) = 0;