Home
last modified time | relevance | path

Searched refs:ShuffleDatasetBase (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc42 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 …]