Searched refs:ReshufflingDataset (Results 1 – 1 of 1) sorted by relevance
339 new ReshufflingDataset(ctx, input, buffer_size, seed, seed2, count); in MakeDataset()349 class ReshufflingDataset : public ShuffleDatasetBase { class in tensorflow::__anon090660da0111::ShuffleDatasetOp351 ReshufflingDataset(OpKernelContext* ctx, const DatasetBase* input, in ReshufflingDataset() function in tensorflow::__anon090660da0111::ShuffleDatasetOp::ReshufflingDataset