Searched refs:datasets_ (Results 1 – 3 of 3) sorted by relevance
61 auto& stored_dataset = datasets_[key]; in Put()72 auto& stored_dataset = datasets_[key]; in Get()
71 absl::flat_hash_map<std::string, std::shared_ptr<const DatasetDef>> datasets_;
486 : DatasetBase(DatasetContext(std::move(params))), datasets_(datasets) { in NestedDataset()526 std::vector<DatasetBase*> datasets_; member in tensorflow::data::snapshot_util::Reader::NestedDataset539 const int64 num_datasets = dataset()->datasets_.size(); in GetNextInternal()545 StoreDatasetInVariantTensor(dataset()->datasets_[index_], &tensor)); in GetNextInternal()