Searched refs:NextStepId (Results 1 – 10 of 10) sorted by relevance
80 TEST_F(RpcCollectiveExecutorMgrTest, NextStepId) { in TEST_F() argument81 int64 x = cme_->NextStepId(7); in TEST_F()94 x = cme_->NextStepId(7); in TEST_F()97 EXPECT_EQ(x, cme_->NextStepId(7)); in TEST_F()98 EXPECT_EQ(x, cme_->NextStepId(7)); in TEST_F()101 EXPECT_EQ(x, cme_->NextStepId(7)); in TEST_F()104 int64 y = cme_->NextStepId(7); in TEST_F()119 int64 z = cme_->NextStepId(7); in TEST_F()126 int64 x = cme_->NextStepId(3); in TEST_F()128 int64 y = cme_->NextStepId(4); in TEST_F()
54 int64 NextStepId(int64 graph_key) override;
146 int64 RpcCollectiveExecutorMgr::NextStepId(int64 graph_key) { in NextStepId() function in tensorflow::RpcCollectiveExecutorMgr
1483 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()1501 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()
52 int64 NextStepId(int64 graph_key) override { in NextStepId() function
73 EXPECT_EQ(CollectiveExecutor::kInvalidId, cme_->NextStepId(123)); in TEST_F()
106 int64 NextStepId(int64 graph_key) override { in NextStepId() function
451 int64 NextStepId();
462 int64 XlaCompiler::NextStepId() { return next_step_id_++; } in NextStepId() function in tensorflow::XlaCompiler1128 flib_runtime_, NextStepId())); in CompileGraph()
200 virtual int64 NextStepId(int64 graph_key) = 0;