| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/samplers/ |
| D | bindings.cc | 59 .def(py::init([](int64_t num_samples, py::object sampler) { in __anon56c245d60502()
70 .def(py::init([](int64_t num_val, bool shuffle, int64_t num_samples) { in __anon56c245d60702()
81 … .def(py::init([](bool replacement, int64_t num_samples, bool reshuffle_each_epoch) { in __anon56c245d60902()
92 .def(py::init([](int64_t start_index, int64_t num_samples) { in __anon56c245d60b02()
103 .def(py::init([](std::vector<int64_t> indices, int64_t num_samples) { in __anon56c245d60d02()
114 .def(py::init([](std::vector<int64_t> indices, int64_t num_samples) { in __anon56c245d60f02()
125 … .def(py::init([](std::vector<double> weights, int64_t num_samples, bool replacement) { in __anon56c245d61102()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/ |
| D | samplers.cc | 42 …ampler::DistributedSampler(int64_t num_shards, int64_t shard_id, bool shuffle, int64_t num_samples, in DistributedSampler()
63 PKSampler::PKSampler(int64_t num_val, bool shuffle, int64_t num_samples) in PKSampler()
76 RandomSampler::RandomSampler(bool replacement, int64_t num_samples) in RandomSampler()
89 SequentialSampler::SequentialSampler(int64_t start_index, int64_t num_samples) in SequentialSampler()
102 SubsetSampler::SubsetSampler(const std::vector<int64_t> &indices, int64_t num_samples) in SubsetSampler()
115 SubsetRandomSampler::SubsetRandomSampler(const std::vector<int64_t> &indices, int64_t num_samples) in SubsetRandomSampler()
128 …Sampler::WeightedRandomSampler(const std::vector<double> &weights, int64_t num_samples, bool repla… in WeightedRandomSampler()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/ |
| D | test_datasets_sharding.py | 27 def sharding_config(num_shards, shard_id, num_samples, shuffle, class_index, repeat_cnt=1): argument
66 def sharding_config(num_shards, shard_id, num_samples, repeat_cnt=1): argument
108 def sharding_config(num_shards, shard_id, num_samples, repeat_cnt=1): argument
157 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
187 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
220 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
245 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
269 def sharding_config(num_shards, shard_id, num_samples, shuffle, repeat_cnt=1): argument
|
| D | test_pyfunc_multiprocess_autotune.py | 29 def create_pyfunc_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None): argument
60 def create_pyop_cop_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None): argument
105 def create_mixed_map_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None… argument
133 def create_per_batch_map_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=… argument
162 def create_mp_dataset(batch_size=32, repeat_size=1, num_parallel_workers=1, num_samples=None): argument
|
| D | test_sampler.py | 36 def test_config(num_samples, num_repeats=None): argument
65 def test_config(replacement, num_samples, num_repeats): argument
95 def test_config(replacement, num_samples, num_repeats, validate): argument
137 def __init__(self, num_samples=None): argument
200 def test_config(start_index, num_samples): argument
228 def test_config(indices, num_samples=None, exception_msg=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/bindings/mindrecord/include/ |
| D | bindings.cc | 41 … int64_t num_samples) { self->SetNumSamples(num_samples); }); in __anond6dedc9e0102()
55 … .def(py::init([](int64_t kVal, const std::string &kColumn, bool shuffle, int64_t num_samples) { in __anond6dedc9e0502()
77 .def(py::init([](int64_t num_samples, int64_t start_index) { in __anond6dedc9e0802()
86 .def(py::init([](int64_t num_samples, bool replacement, bool reshuffle_each_epoch) { in __anond6dedc9e0a02()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/mindrecord/meta/ |
| D | shard_pk_sample.cc | 21 …Sample::ShardPkSample(const std::string &category_field, int64_t num_elements, int64_t num_samples) in ShardPkSample()
27 int64_t num_samples) in ShardPkSample()
31 uint32_t seed, int64_t num_samples) in ShardPkSample()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/engine/ |
| D | datasets_text.py | 130 def __init__(self, dataset_dir, usage=None, num_samples=None, argument
234 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=S… argument
477 …def __init__(self, dataset_files, task='AFQMC', usage='train', num_samples=None, num_parallel_work… argument
577 …def __init__(self, dataset_dir, usage=None, num_samples=None, shuffle=Shuffle.GLOBAL, num_shards=N… argument
682 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=S… argument
773 def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=True, argument
896 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
1030 …num_samples=None, shuffle=Shuffle.GLOBAL, num_shards=None, shard_id=None, num_parallel_workers=Non… argument
1139 …def __init__(self, dataset_dir, usage=None, language_pair=None, num_samples=None, shuffle=Shuffle.… argument
1247 def __init__(self, dataset_dir, usage=None, language_pair=None, num_samples=None, argument
[all …]
|
| D | samplers.py | 29 def select_sampler(num_samples, input_sampler, shuffle, num_shards, shard_id): argument
99 def __init__(self, num_samples=None): argument
251 def __init__(self, num_samples=None): argument
271 def _handshake(self, ds_size, num_samples): argument
368 def __init__(self, num_shards, shard_id, shuffle=True, num_samples=None, offset=-1): argument
468 …def __init__(self, num_val, num_class=None, shuffle=False, class_column='label', num_samples=None): argument
549 def __init__(self, replacement=False, num_samples=None): argument
616 def __init__(self, start_index=None, num_samples=None): argument
686 def __init__(self, indices, num_samples=None): argument
819 def __init__(self, sampler, num_samples=None): argument
[all …]
|
| D | datasets_vision.py | 226 def __init__(self, dataset_dir, target_type=None, num_samples=None, num_parallel_workers=1, argument
362 …def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=None, decode=… argument
506 … extensions=None, num_samples=None, num_shards=None, shard_id=None, cache=None, decrypt=None): argument
629 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
737 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
889 … __init__(self, dataset_dir, usage="train", quality_mode="fine", task="instance", num_samples=None, argument
1096 …def __init__(self, dataset_dir, annotation_file, task="Detection", num_samples=None, num_parallel_… argument
1299 def __init__(self, dataset_dir, usage="train", downgrade="bicubic", scale=2, num_samples=None, argument
1418 def __init__(self, dataset_dir, name, usage=None, num_samples=None, num_parallel_workers=None, argument
1488 …t__(self, num_images=1000, image_size=(224, 224, 3), num_classes=10, base_seed=0, num_samples=None, argument
[all …]
|
| D | datasets_audio.py | 144 …def __init__(self, dataset_dir, name=None, num_samples=None, num_parallel_workers=None, shuffle=No… argument
256 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
389 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
510 def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=None, argument
614 …def __init__(self, dataset_dir, usage=None, num_samples=None, num_parallel_workers=None, shuffle=N… argument
805 def __init__(self, dataset_dir, release, usage=None, extensions=None, num_samples=None, argument
903 def __init__(self, dataset_dir, num_samples=None, num_parallel_workers=None, shuffle=None, argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/ |
| D | weighted_random_sampler_test.cc | 48 uint64_t num_samples = 100; in TEST_F() local
80 uint64_t num_samples = 100; in TEST_F() local
118 uint64_t num_samples = 100; in TEST_F() local
153 uint64_t num_samples = 100; in TEST_F() local
199 uint64_t num_samples = 1000000; in TEST_F() local
246 uint64_t num_samples = 1000000; in TEST_F() local
|
| D | distributed_sampler_test.cc | 48 uint64_t num_samples = 7; in TEST_F() local
75 uint64_t num_samples = 7; in TEST_F() local
103 uint64_t num_samples = 2; in TEST_F() local
|
| D | subset_sampler_test.cc | 46 int64_t num_samples = 0; in TEST_F() local
76 int64_t num_samples = 0; in TEST_F() local
110 int64_t num_samples = 0; in TEST_F() local
|
| D | subset_random_sampler_test.cc | 46 int64_t num_samples = 0; in TEST_F() local
76 int64_t num_samples = 0; in TEST_F() local
109 int64_t num_samples = 0; in TEST_F() local
|
| D | stand_alone_samplers_test.cc | 65 int64_t num_samples = 0; in TEST_F() local
90 int64_t num_samples = 0; in TEST_F() local
124 int64_t num_samples = 0; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | histogram_fixed_width_impl.cu | 25 cudaError_t HistogramFixedWidthKernel(int num_samples, const T *d_samples, const double *d_levels, … in HistogramFixedWidthKernel()
39 cudaError_t CalHistogramFixedWidth(int num_samples, const T *d_samples, const double *d_levels, int… in CalHistogramFixedWidth()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/samplers/ |
| D | subset_random_sampler_ir.cc | 34 SubsetRandomSamplerObj::SubsetRandomSamplerObj(std::vector<int64_t> indices, int64_t num_samples) in SubsetRandomSamplerObj()
68 Status SubsetRandomSamplerObj::from_json(nlohmann::json json_obj, int64_t num_samples, in from_json()
|
| D | sequential_sampler_ir.cc | 34 SequentialSamplerObj::SequentialSamplerObj(int64_t start_index, int64_t num_samples) in SequentialSamplerObj()
65 Status SequentialSamplerObj::from_json(nlohmann::json json_obj, int64_t num_samples, in from_json()
|
| D | subset_sampler_ir.cc | 34 SubsetSamplerObj::SubsetSamplerObj(std::vector<int64_t> indices, int64_t num_samples) in SubsetSamplerObj()
76 Status SubsetSamplerObj::from_json(nlohmann::json json_obj, int64_t num_samples, std::shared_ptr<Sa… in from_json()
|
| D | pk_sampler_ir.cc | 34 PKSamplerObj::PKSamplerObj(int64_t num_val, bool shuffle, int64_t num_samples) in PKSamplerObj()
64 Status PKSamplerObj::from_json(nlohmann::json json_obj, int64_t num_samples, std::shared_ptr<Sample… in from_json()
|
| D | random_sampler_ir.cc | 34 RandomSamplerObj::RandomSamplerObj(bool replacement, int64_t num_samples, bool reshuffle_each_epoch) in RandomSamplerObj()
60 Status RandomSamplerObj::from_json(nlohmann::json json_obj, int64_t num_samples, std::shared_ptr<Sa… in from_json()
|
| D | weighted_random_sampler_ir.cc | 25 WeightedRandomSamplerObj::WeightedRandomSamplerObj(std::vector<double> weights, int64_t num_samples… in WeightedRandomSamplerObj()
69 Status WeightedRandomSamplerObj::from_json(nlohmann::json json_obj, int64_t num_samples, in from_json()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/ |
| D | bindings.cc | 113 … .def(py::init([](const std::string &dataset_dir, const std::string &usage, int64_t num_samples, in __anon3b221fe00302()
114 int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon3b221fe00302()
188 … int64_t num_samples, int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon3b221fe00f02()
239 … int64_t num_samples, int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon3b221fe01702()
252 int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon3b221fe01902()
289 int32_t num_shards, int32_t shard_id) { in __anon3b221fe01f02()
426 … const std::vector<std::string> &language_pair, int64_t num_samples, in __anon3b221fe03402()
427 int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon3b221fe03402()
562 int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon3b221fe04902()
587 int32_t shuffle, int32_t num_shards, int32_t shard_id) { in __anon3b221fe04d02()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/ |
| D | sequential_sampler.cc | 25 SequentialSamplerRT::SequentialSamplerRT(int64_t start_index, int64_t num_samples, int64_t samples_… in SequentialSamplerRT()
123 …int64_t num_samples = (num_samples_ > 0) ? std::min(child_num_rows, num_samples_) : child_num_rows; in CalculateNumSamples() local
|