Home
last modified time | relevance | path

Searched defs:MemoryDatasetBase (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dcache_dataset_ops.h26 class MemoryDatasetBase; variable
H A Dcache_dataset_ops.cc724 class CacheDatasetOp::MemoryDatasetBase : public DatasetBase { class in tensorflow::data::CacheDatasetOp
726 explicit MemoryDatasetBase(OpKernelContext* ctx, const DatasetBase* input, in MemoryDatasetBase() function in tensorflow::data::CacheDatasetOp::MemoryDatasetBase