Searched refs:BatchDataset (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/ |
| D | datasets.h | 58 class BatchDataset; variable
176 std::shared_ptr<BatchDataset> Batch(int32_t batch_size, bool drop_remainder = false);
413 class BatchDataset : public Dataset {
415 BatchDataset(std::shared_ptr<Dataset> input, int32_t batch_size, bool drop_remainder = false);
416 ~BatchDataset() = default;
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 496 …return BatchDataset(self, batch_size, drop_remainder, num_parallel_workers, per_batch_map, input_c…
2130 class BatchDataset(Dataset): class
2169 if BatchDataset._is_ancestor_of_repeat(input_dataset):
2172 BatchDataset._update_batch_size_for_syncwait(input_dataset, batch_size)
2217 flag = flag | BatchDataset._is_ancestor_of_repeat(input_dataset)
2232 BatchDataset._update_batch_size_for_syncwait(input_dataset, batch_size)
2452 if isinstance(dataset, BatchDataset):
3028 if isinstance(child, BatchDataset):
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 63 class BatchDataset; variable
196 std::shared_ptr<BatchDataset> Batch(int32_t batch_size, bool drop_remainder = false);
600 class BatchDataset : public Dataset {
610 BatchDataset(std::shared_ptr<Dataset> input, int32_t batch_size, bool drop_remainder = false);
613 ~BatchDataset() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 398 BatchDataset::BatchDataset(std::shared_ptr<Dataset> input, int32_t batch_size, bool drop_remainder)… in BatchDataset() function in mindspore::dataset::BatchDataset
626 std::shared_ptr<BatchDataset> Dataset::Batch(int32_t batch_size, bool drop_remainder) { in Batch()
627 return std::make_shared<BatchDataset>(shared_from_this(), batch_size, drop_remainder); in Batch()
|