Home
last modified time | relevance | path

Searched refs:GetNextSample (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/dataset/
Dweighted_random_sampler_test.cc60 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
71 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
89 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
106 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
124 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
135 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
159 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
171 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
199 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
209 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
[all …]
Ddistributed_sampler_test.cc57 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
66 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
81 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
91 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
106 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
116 ASSERT_EQ(m_sampler.GetNextSample(&row), Status::OK()); in TEST_F()
Dsubset_sampler_test.cc51 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
63 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
80 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
91 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
110 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
124 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
138 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
Dsubset_random_sampler_test.cc51 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
63 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
80 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
90 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
109 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
122 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
135 ASSERT_EQ(sampler.GetNextSample(&row), Status::OK()); in TEST_F()
Dstand_alone_samplers_test.cc67 sampler->GetNextSample(&sample_row); in TEST_F()
90 sampler->GetNextSample(&sample_row); in TEST_F()
93 sampler->GetNextSample(&sample_row); in TEST_F()
97 sampler->GetNextSample(&sample_row); in TEST_F()
100 sampler->GetNextSample(&sample_row); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dmappable_leaf_op.cc32 RETURN_IF_NOT_OK(sampler_->GetNextSample(&sample_row)); in operator ()()
45 RETURN_IF_NOT_OK(sampler_->GetNextSample(&sample_row)); in operator ()()
70 RETURN_IF_NOT_OK(sampler_->GetNextSample(&sample_row)); in operator ()()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/
Dpython_sampler.cc26 Status PythonSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::PythonSamplerRT
31 RETURN_IF_NOT_OK(child_[0]->GetNextSample(&child_ids_)); in GetNextSample()
Dsubset_sampler.cc59 Status SubsetSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::SubsetSamplerRT
65 RETURN_IF_NOT_OK(child_[0]->GetNextSample(&child_ids_)); in GetNextSample()
Dsequential_sampler.cc26 Status SequentialSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::SequentialSamplerRT
35 RETURN_IF_NOT_OK(child_[0]->GetNextSample(&child_ids_)); in GetNextSample()
Drandom_sampler.cc33 Status RandomSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::RandomSamplerRT
41 RETURN_IF_NOT_OK(child_[0]->GetNextSample(&child_ids_)); in GetNextSample()
Dpk_sampler.cc71 Status PKSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::PKSamplerRT
80 RETURN_IF_NOT_OK(child_[0]->GetNextSample(&child_ids_)); in GetNextSample()
Dweighted_random_sampler.cc115 Status WeightedRandomSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::WeightedRandomSamplerRT
134 RETURN_IF_NOT_OK(child_[0]->GetNextSample(&child_ids_)); in GetNextSample()
Dsampler.cc98 RETURN_IF_NOT_OK(GetNextSample(&sample_row)); in GetAllIdsThenReset()
105 RETURN_IF_NOT_OK(GetNextSample(&sample_row)); in GetAllIdsThenReset()
Dpython_sampler.h51 Status GetNextSample(TensorRow *out) override;
Ddistributed_sampler.cc94 Status DistributedSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::DistributedSamplerRT
117 RETURN_IF_NOT_OK(child_[0]->GetNextSample(&child_ids_)); in GetNextSample()
Dmind_record_sampler.h42 Status GetNextSample(TensorRow *out) override;
Dsequential_sampler.h50 Status GetNextSample(TensorRow *out) override;
Dsubset_sampler.h52 Status GetNextSample(TensorRow *out) override;
Drandom_sampler.h44 Status GetNextSample(TensorRow *out) override;
Dweighted_random_sampler.h54 Status GetNextSample(TensorRow *out) override;
Dpk_sampler.h44 Status GetNextSample(TensorRow *out) override;
Ddistributed_sampler.h53 Status GetNextSample(TensorRow *out) override;
Dmind_record_sampler.cc25 Status MindRecordSamplerRT::GetNextSample(TensorRow *out) { in GetNextSample() function in mindspore::dataset::MindRecordSamplerRT
Dsampler.h79 virtual Status GetNextSample(TensorRow *out) = 0;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dcache_lookup_op.h47 Status GetNextSample(TensorRow *out) override;

12