| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | weighted_random_sampler_test.cc | 114 uint64_t samples_per_tensor = 10; in TEST_F() local
118 WeightedRandomSamplerRT m_sampler(weights, num_samples, true, samples_per_tensor); in TEST_F()
138 ASSERT_EQ(epoch, (num_samples + samples_per_tensor - 1) / samples_per_tensor); in TEST_F()
146 uint64_t samples_per_tensor = 10; in TEST_F() local
153 WeightedRandomSamplerRT m_sampler(weights, num_samples, false, samples_per_tensor); in TEST_F()
181 ASSERT_EQ(epoch, (num_samples + samples_per_tensor - 1) / samples_per_tensor); in TEST_F()
|
| D | subset_sampler_test.cc | 69 int64_t samples_per_tensor = 10; in TEST_F() local
72 SubsetSamplerRT sampler(input, num_samples, samples_per_tensor); in TEST_F()
94 ASSERT_EQ(epoch, (total_samples + samples_per_tensor - 1) / samples_per_tensor); in TEST_F()
|
| D | subset_random_sampler_test.cc | 69 int64_t samples_per_tensor = 10; in TEST_F() local
72 SubsetRandomSamplerRT sampler(input, num_samples, samples_per_tensor); in TEST_F()
93 ASSERT_EQ(epoch, (total_samples + samples_per_tensor - 1) / samples_per_tensor); in TEST_F()
|
| D | image_folder_op_test.cc | 240 int64_t samples_per_tensor = 10; in TEST_F() local
245 std::make_shared<WeightedRandomSamplerRT>(weights, num_samples, true, samples_per_tensor); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/ |
| D | subset_random_sampler.cc | 29 int64_t samples_per_tensor) in SubsetRandomSamplerRT() argument
30 : SubsetSamplerRT(indices, num_samples, samples_per_tensor) {} in SubsetRandomSamplerRT()
|
| D | mind_record_sampler.cc | 22 …rdSamplerRT::MindRecordSamplerRT(mindrecord::ShardReader *shard_reader, int64_t samples_per_tensor) in MindRecordSamplerRT() argument
23 …: SamplerRT(0, samples_per_tensor), shard_reader_(shard_reader), sample_ids_(nullptr), next_id_(0)… in MindRecordSamplerRT()
|
| D | python_sampler.cc | 21 …T::PythonSamplerRT(int64_t num_samples, py::object py_sampler_instance, int64_t samples_per_tensor) in PythonSamplerRT() argument
22 : SamplerRT(num_samples, samples_per_tensor), in PythonSamplerRT()
|
| D | subset_sampler.cc | 21 …bsetSamplerRT(const std::vector<int64_t> &indices, int64_t num_samples, int64_t samples_per_tensor) in SubsetSamplerRT() argument
22 : SamplerRT(num_samples, samples_per_tensor), indices_(indices), sample_id_(0) {} in SubsetSamplerRT()
|
| D | sequential_sampler.cc | 23 …amplerRT::SequentialSamplerRT(int64_t start_index, int64_t num_samples, int64_t samples_per_tensor) in SequentialSamplerRT() argument
24 …: SamplerRT(num_samples, samples_per_tensor), current_id_(start_index), start_index_(start_index),… in SequentialSamplerRT()
|
| D | random_sampler.cc | 25 int64_t samples_per_tensor) in RandomSamplerRT() argument
26 : SamplerRT(num_samples, samples_per_tensor), in RandomSamplerRT()
|
| D | pk_sampler.cc | 23 …plerRT::PKSamplerRT(int64_t num_val, bool shuffle, int64_t num_samples, int64_t samples_per_tensor) in PKSamplerRT() argument
24 : SamplerRT(num_samples, samples_per_tensor), in PKSamplerRT()
|
| D | weighted_random_sampler.cc | 31 bool replacement, int64_t samples_per_tensor) in WeightedRandomSamplerRT() argument
32 …: SamplerRT(num_samples, samples_per_tensor), weights_(weights), replacement_(replacement), sample… in WeightedRandomSamplerRT()
|
| D | sampler.cc | 36 SamplerRT::SamplerRT(int64_t num_samples, int64_t samples_per_tensor) in SamplerRT() argument
39 samples_per_tensor_(samples_per_tensor), in SamplerRT()
|
| D | python_sampler.h | 34 int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | subset_random_sampler.h | 37 std::int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | mind_record_sampler.h | 34 int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | sequential_sampler.h | 34 int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | subset_sampler.h | 36 std::int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | random_sampler.h | 35 int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | weighted_random_sampler.h | 38 int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | pk_sampler.h | 36 int64_t samples_per_tensor = std::numeric_limits<int64_t>::max());
|
| D | sampler.h | 67 SamplerRT(int64_t num_samples, int64_t samples_per_tensor);
|