Home
last modified time | relevance | path

Searched refs:indices_ (Results 1 – 25 of 37) sorted by relevance

12

/third_party/boost/boost/histogram/
Dindexed.hpp130 return iter_.indices_[d].idx; in index()
135 assert(iter_.indices_.hist_); in indices()
136 return {iter_.indices_.begin(), iter_.indices_.end()}; in indices()
143 assert(iter_.indices_.hist_);
144 return iter_.indices_.hist_->axis(std::integral_constant<unsigned, N>())
151 assert(iter_.indices_.hist_); in bin()
152 return iter_.indices_.hist_->axis(d).bin(index(d)); in bin()
161 assert(iter_.indices_.hist_); in density()
164 iter_.indices_.hist_->for_each_axis([&](const auto& a) { in density()
242 assert(iter_ < indices_.hist_->end()); in operator ++()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/
Dsubset_sampler.cc22 : SamplerRT(num_samples, samples_per_tensor), indices_(indices), sample_id_(0) {} in SubsetSamplerRT()
34 if (num_samples_ == 0 || num_samples_ > static_cast<int64_t>(indices_.size())) { in InitSampler()
35 num_samples_ = static_cast<int64_t>(indices_.size()); in InitSampler()
82 if (indices_[sample_id_] >= num_rows_ || indices_[sample_id_] < 0) { in GetNextSample()
83 std::string err_msg = "Sample ID (" + std::to_string(indices_[sample_id_]) + in GetNextSample()
88 int64_t sampled_id = ((indices_[sample_id_] % num_rows_) + num_rows_) % num_rows_; in GetNextSample()
117 args["indices"] = indices_; in to_json()
131 res = std::min(res, static_cast<int64_t>(indices_.size())); in CalculateNumSamples()
Dsubset_random_sampler.cc43 std::shuffle(indices_.begin(), indices_.end(), rand_gen_); in InitSampler()
52 std::shuffle(indices_.begin(), indices_.end(), rand_gen_); in ResetSampler()
Dsubset_sampler.h72 std::vector<int64_t> indices_;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_optimizer_cpu_kernel.h33 T *indices_{nullptr};
74 T *indices_; member
147 MS_EXCEPTION_IF_NULL(sparse_grad->indices_); in CalculateEachBucketSize()
154 T index = sparse_grad->indices_[i]; in CalculateEachBucketSize()
191 segments[i]->indices_ = input_grad->indices_ + current_indices_offset; in SplitAndCalculateSegmentBucketSize()
209 MS_EXCEPTION_IF_NULL(segment->indices_); in CopySegmentIndicesToBucket()
215 T index = segment->indices_[i]; in CopySegmentIndicesToBucket()
219 buckets[bucket_id]->indices_[bucket_index] = index; in CopySegmentIndicesToBucket()
235 MS_EXCEPTION_IF_NULL(param.output_grad_->indices_); in GatherSegmentIndicesToOutputBucket()
252 buckets[i]->indices_ = param.output_grad_->indices_ + current_indices_offset; in GatherSegmentIndicesToOutputBucket()
[all …]
Dscatter_nd_cpu_kernel.cc32 S *indices = params->indices_; in Compute()
107 params.indices_ = reinterpret_cast<S *>(inputs[0]->addr); in Launch()
Dscatter_nd_update_cpu_kernel.cc33 int *indices = params->indices_; in Compute()
121 params.indices_ = reinterpret_cast<int *>(inputs[1]->addr); in LaunchKernel()
Dscatter_nd_update_cpu_kernel.h31 int *indices_{nullptr};
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/meta/
Dshard_sample.cc30 indices_({}), in ShardSample()
39 indices_({}), in ShardSample()
48 indices_({}), in ShardSample()
57 indices_(indices), in ShardSample()
78 return indices_.size(); in GetNumSamples()
88 for (int i = 0; i < indices_.size(); ++i) { in UpdateTasks()
89 int index = ((indices_[i] % total_no) + total_no) % total_no; in UpdateTasks()
146 CHECK_FAIL_RETURN_UNEXPECTED(indices_.size() <= static_cast<size_t>(total_no), in Execute()
147 "Invalid input, indices size: " + std::to_string(indices_.size()) + in Execute()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/samplers/
Dsubset_sampler_ir.cc35 : indices_(std::move(indices)), num_samples_(num_samples) {} in SubsetSamplerObj()
51 *sampler = std::make_shared<dataset::SubsetSamplerRT>(indices_, num_samples_); in SamplerBuild()
60 auto mind_sampler = std::make_shared<mindrecord::ShardSample>(indices_); in BuildForMindDataset()
69 args["indices"] = indices_; in to_json()
87 auto sampler = std::make_shared<SubsetSamplerObj>(indices_, num_samples_); in SamplerCopy()
Dsubset_random_sampler_ir.cc42 *sampler = std::make_shared<dataset::SubsetRandomSamplerRT>(indices_, num_samples_); in SamplerBuild()
51 auto mind_sampler = std::make_shared<mindrecord::ShardSample>(indices_, GetSeed()); in BuildForMindDataset()
61 args["indices"] = indices_; in to_json()
80 auto sampler = std::make_shared<SubsetRandomSamplerObj>(indices_, num_samples_); in SamplerCopy()
Dsubset_sampler_ir.h70 const std::vector<int64_t> indices_;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/
Dfused_push_weight_kernel.h37 … : server_num_(0), indices_({}), weight_full_names_({}), fl_iteration_(0), total_iteration_(0) {} in FusedPushWeightKernel()
123 indices_ = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, kAttrIndex); in Init()
125 … MS_LOG(INFO) << "Weight full name as key " << weight_full_names_ << ", key index is " << indices_ in Init()
127 if (server_num_ == 0 || weight_full_names_.empty() || indices_.empty()) { in Init()
167 std::vector<int64_t> indices_; variable
Dfused_pull_weight_kernel.h39 … : server_num_(0), indices_({}), weight_full_names_({}), fl_iteration_(0), total_iteration_(0) {} in FusedPullWeightKernel()
136 indices_ = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, kAttrIndex); in Init()
138 … MS_LOG(INFO) << "Weight full name as key " << weight_full_names_ << ", key index is " << indices_ in Init()
140 if (server_num_ == 0 || weight_full_names_.empty() || indices_.empty()) { in Init()
195 std::vector<int64_t> indices_; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dsamplers.cc73 : indices_(indices), num_samples_(num_samples) {} in SubsetSampler()
76 return std::make_shared<SubsetSamplerObj>(indices_, num_samples_); in Parse()
84 return std::make_shared<SubsetRandomSamplerObj>(indices_, num_samples_); in Parse()
/third_party/skia/third_party/externals/tint/src/sem/
Dmember_accessor_expression.h96 const std::vector<uint32_t>& Indices() const { return indices_; } in Indices()
99 std::vector<uint32_t> const indices_;
Dmember_accessor_expression.cc48 : Base(declaration, type, statement), indices_(std::move(indices)) {} in Swizzle()
/third_party/mindspore/mindspore/core/abstract/
Dabstract_value.cc1263 MS_EXCEPTION_IF_NULL(indices_); in Clone()
1266 auto indices_clone = indices_->Clone(); in Clone()
1285 MS_EXCEPTION_IF_NULL(indices_); in Broaden()
1288 auto indices_clone = indices_->Clone(); in Broaden()
1308 MS_EXCEPTION_IF_NULL(indices_); in BroadenWithShape()
1311 auto indices_clone = indices_->Clone(); in BroadenWithShape()
1330 MS_EXCEPTION_IF_NULL(indices_); in ToString()
1336 << ", indices: " << indices_->ToString() << ", values" << values_->ToString() in ToString()
1355 MS_EXCEPTION_IF_NULL(indices_); in Clone()
1358 auto indices_clone = indices_->Clone(); in Clone()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dtensor_helpers.cc57 for (size_t k = 0; k < slice_list[curr_ind].indices_.size(); k++) { in IndexGeneratorHelper()
58 (*numbers)[curr_ind] = slice_list[curr_ind].indices_[k]; in IndexGeneratorHelper()
/third_party/mindspore/tests/ut/cpp/kernel/cpu/
Dsparse_optimizer_cpu_kernel_test.cc69 EXPECT_EQ(unique_grad.indices_[i], expect_indices[i]); in TEST_F()
122 EXPECT_EQ(unique_grad.indices_[i], expect_indices[i]); in TEST_F()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/
Dcommon_utils_test.cc69 EXPECT_EQ(unique_grad.indices_[i], expect_indices[i]); in TEST_F()
122 EXPECT_EQ(unique_grad.indices_[i], expect_indices[i]); in TEST_F()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dgather_fp32_coder.h38 int32_t *indices_{nullptr};
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/
Dgather_tensorrt.h38 mindspore::MSTensor indices_; variable
/third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/
Dshard_sample.h60 std::vector<int64_t> indices_;
/third_party/vk-gl-cts/framework/opengl/
DgluDrawUtil.hpp174 PrimitiveList (PrimitiveType type_, int numElements_, IndexType indexType_, const void* indices_) in PrimitiveList()
178 , indices (indices_) in PrimitiveList()

12