Searched refs:DatasetFromId (Results 1 – 4 of 4) sorted by relevance
240 TF_RETURN_IF_ERROR(state_.DatasetFromId(task->job->dataset_id, dataset)); in WorkerHeartbeat()301 TF_RETURN_IF_ERROR(state_.DatasetFromId(request->dataset_id(), dataset)); in GetDatasetDef()353 TF_RETURN_IF_ERROR(state_.DatasetFromId(dataset_id, dataset)); in MakeSplitProvider()667 TF_RETURN_IF_ERROR(state_.DatasetFromId(task->job->dataset_id, dataset)); in AssignTask()874 TF_RETURN_IF_ERROR(state_.DatasetFromId(dataset_id, dataset)); in GetDatasetDef()
171 Status DatasetFromId(int64 id, std::shared_ptr<const Dataset>& dataset) const;
140 TF_EXPECT_OK(state.DatasetFromId(id, dataset)); in TEST()148 Status s = state.DatasetFromId(0, dataset); in TEST()
220 Status DispatcherState::DatasetFromId( in DatasetFromId() function in tensorflow::data::DispatcherState