Searched refs:MemoryDataset (Results 1 – 1 of 1) sorted by relevance
43 *output = new MemoryDataset(ctx, input); in MakeDataset()568 class MemoryDataset : public DatasetBase { class in tensorflow::data::__anon80340b800111::CacheDatasetOp570 explicit MemoryDataset(OpKernelContext* ctx, const DatasetBase* input) in MemoryDataset() function in tensorflow::data::__anon80340b800111::CacheDatasetOp::MemoryDataset575 ~MemoryDataset() override { input_->Unref(); } in ~MemoryDataset()688 class MemoryIterator : public DatasetIterator<MemoryDataset> {691 : DatasetIterator<MemoryDataset>(params) {} in MemoryIterator()805 class MemoryWriterIterator : public DatasetIterator<MemoryDataset> {808 : DatasetIterator<MemoryDataset>(params), cache_(cache) { in MemoryWriterIterator()869 class MemoryReaderIterator : public DatasetIterator<MemoryDataset> {872 : DatasetIterator<MemoryDataset>(params), cache_(cache), index_(0) { in MemoryReaderIterator()