Searched refs:ShuffleDatasetBase (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 82 class ShuffleDatasetOpBase::ShuffleDatasetBase : public DatasetBase { class in tensorflow::data::ShuffleDatasetOpBase
84 ShuffleDatasetBase(OpKernelContext* ctx, const DatasetBase* input, in ShuffleDatasetBase() function in tensorflow::data::ShuffleDatasetOpBase::ShuffleDatasetBase
98 ~ShuffleDatasetBase() override { input_->Unref(); } in ~ShuffleDatasetBase()
144 class Iterator : public DatasetIterator<ShuffleDatasetBase> {
147 : DatasetIterator<ShuffleDatasetBase>(params), in Iterator()
441 class ShuffleDatasetOp::Dataset : public ShuffleDatasetBase {
446 : ShuffleDatasetBase(ctx, input, buffer_size, manager->get(), count), in Dataset()
499 class ShuffleDatasetOp::DatasetV2 : public ShuffleDatasetBase {
504 : ShuffleDatasetBase(ctx, input, buffer_size, manager->get(), count), in DatasetV2()
553 class ShuffleDatasetOp::DatasetV3 : public ShuffleDatasetBase {
[all …]
|
| D | shuffle_dataset_op.h | 37 class ShuffleDatasetBase;
|