Searched refs:PKSamplerRT (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/ |
| D | pk_sampler.cc | 23 PKSamplerRT::PKSamplerRT(int64_t num_val, bool shuffle, int64_t num_samples, int64_t samples_per_te… in PKSamplerRT() function in mindspore::dataset::PKSamplerRT
30 Status PKSamplerRT::InitSampler() { in InitSampler()
71 Status PKSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample()
107 Status PKSamplerRT::ResetSampler() { in ResetSampler()
119 Status PKSamplerRT::HandshakeRandomAccessOp(const RandomAccessOp *op) { in HandshakeRandomAccessOp()
126 void PKSamplerRT::SamplerPrint(std::ostream &out, bool show_all) const { in SamplerPrint()
135 Status PKSamplerRT::to_json(nlohmann::json *out_json) { in to_json()
|
| D | pk_sampler.h | 29 class PKSamplerRT : public SamplerRT { // NOT YET FINISHED
35 PKSamplerRT(int64_t num_val, bool shuffle, int64_t num_samples,
39 ~PKSamplerRT() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/samplers/ |
| D | pk_sampler_ir.cc | 78 *sampler = std::make_shared<dataset::PKSamplerRT>(num_val_, shuffle_, num_samples_); in SamplerBuild()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | image_folder_op_test.cc | 335 std::shared_ptr<SamplerRT> sampler = std::make_shared<PKSamplerRT>(3, false, num_samples, 4); in TEST_F()
|