Searched refs:CreateSamplerTensor (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/ |
| D | distributed_sampler.cc | 110 RETURN_IF_NOT_OK(CreateSamplerTensor(&sample_ids, 1)); in GetNextSample()
121 RETURN_IF_NOT_OK(CreateSamplerTensor(&sample_ids, samples_per_tensor_)); in GetNextSample()
|
| D | mind_record_sampler.cc | 35 RETURN_IF_NOT_OK(CreateSamplerTensor(&sampleIdsTensor, last_id - next_id_)); in GetNextSample()
|
| D | sampler.h | 128 Status CreateSamplerTensor(std::shared_ptr<Tensor> *sample_ids, int64_t num_elements);
|
| D | subset_sampler.cc | 77 RETURN_IF_NOT_OK(CreateSamplerTensor(&outputIds, last_id - sample_id_)); in GetNextSample()
|
| D | sequential_sampler.cc | 45 RETURN_IF_NOT_OK(CreateSamplerTensor(&sampleIds, num_elements)); in GetNextSample()
|
| D | random_sampler.cc | 46 RETURN_IF_NOT_OK(CreateSamplerTensor(&sampleIds, last_id - next_id_)); in GetNextSample()
|
| D | pk_sampler.cc | 85 RETURN_IF_NOT_OK(CreateSamplerTensor(&sample_ids, last_id - next_id_)); in GetNextSample()
|
| D | weighted_random_sampler.cc | 146 RETURN_IF_NOT_OK(CreateSamplerTensor(&outputIds, last_id - sample_id_)); in GetNextSample()
|
| D | sampler.cc | 72 Status SamplerRT::CreateSamplerTensor(std::shared_ptr<Tensor> *sample_ids, int64_t num_elements) { in CreateSamplerTensor() function in mindspore::dataset::SamplerRT
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | cache_lookup_op.cc | 81 RETURN_IF_NOT_OK(CreateSamplerTensor(&sample_ts, cache_miss.size())); in GetNextSample()
|