Searched refs:NextAvailableDatasetId (Results 1 – 4 of 4) sorted by relevance
129 int64 id = state.NextAvailableDatasetId(); in TEST()131 EXPECT_EQ(state.NextAvailableDatasetId(), id + 1); in TEST()159 TEST(DispatcherState, NextAvailableDatasetId) { in TEST() argument161 int64 id = state.NextAvailableDatasetId(); in TEST()164 EXPECT_NE(state.NextAvailableDatasetId(), id); in TEST()165 EXPECT_EQ(state.NextAvailableDatasetId(), state.NextAvailableDatasetId()); in TEST()
169 int64 NextAvailableDatasetId() const;
216 int64 DispatcherState::NextAvailableDatasetId() const { in NextAvailableDatasetId() function in tensorflow::data::DispatcherState
410 dataset_id = state_.NextAvailableDatasetId(); in RegisterDataset()