Searched refs:ShuffleDatasetBase (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | shuffle_dataset_op.cc | 42 class ShuffleDatasetBase : public GraphDatasetBase { class in tensorflow::__anon090660da0111::ShuffleDatasetOpBase 44 ShuffleDatasetBase(OpKernelContext* ctx, const DatasetBase* input, in ShuffleDatasetBase() function in tensorflow::__anon090660da0111::ShuffleDatasetOpBase::ShuffleDatasetBase 53 ~ShuffleDatasetBase() override { input_->Unref(); } in ~ShuffleDatasetBase() 64 class Iterator : public DatasetIterator<ShuffleDatasetBase> { 67 : DatasetIterator<ShuffleDatasetBase>(params), in Iterator() 349 class ReshufflingDataset : public ShuffleDatasetBase { 353 : ShuffleDatasetBase(ctx, input, buffer_size, count), in ReshufflingDataset() 373 return std::unique_ptr<IteratorBase>(new ShuffleDatasetBase::Iterator( in MakeIterator() 389 class FixedSeedDataset : public ShuffleDatasetBase { 393 : ShuffleDatasetBase(ctx, input, buffer_size, count), in FixedSeedDataset() [all …]
|