Searched refs:ReshufflingDataset (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 378 new ReshufflingDataset(ctx, input, buffer_size, seed, seed2, count); in MakeDataset()
388 class ReshufflingDataset : public ShuffleDatasetBase { class in tensorflow::data::__anon0b80ed660111::ShuffleDatasetOp
390 ReshufflingDataset(OpKernelContext* ctx, const DatasetBase* input, in ReshufflingDataset() function in tensorflow::data::__anon0b80ed660111::ShuffleDatasetOp::ReshufflingDataset
458 class Iterator : public ShuffleDatasetBase::Iterator<ReshufflingDataset> {
461 : ShuffleDatasetBase::Iterator<ReshufflingDataset>(params, seed, in Iterator()
517 return ShuffleDatasetBase::Iterator<ReshufflingDataset>::SaveInternal( in SaveInternal()
532 ReshufflingDataset>::RestoreInternal(ctx, reader); in RestoreInternal()
|