Home
last modified time | relevance | path

Searched refs:indices_size (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dscatter_functor_impl.cu54 …catterFunc(enum ScatterFunctorType func_type, const size_t &inner_size, const size_t &indices_size, in ScatterFunc() argument
56 const size_t updates_size = inner_size * indices_size; in ScatterFunc()
73 … const size_t &indices_size, const int *indices, const float *updates,
76 … const size_t &indices_size, const int64_t *indices, const float *updates,
79 … const size_t &indices_size, const int *indices, const half *updates, half *input,
82 … const size_t &indices_size, const int64_t *indices, const half *updates,
85 … const size_t &indices_size, const int *indices, const int *updates, int *input,
88 … const size_t &indices_size, const int64_t *indices, const int *updates,
91 const size_t &indices_size, const int *indices,
95 … const size_t &indices_size, const int64_t *indices,
[all …]
Dsparse_apply_proximal_adagrad_impl.cu60 __global__ void SparseApplyProximalAdagradUpdate(const size_t size, const size_t indices_size, cons… in SparseApplyProximalAdagradUpdate() argument
64 const int inner_size = static_cast<int>(size / indices_size); in SparseApplyProximalAdagradUpdate()
84 void CalSparseApplyProximalAdagrad(const size_t size, const size_t indices_size, const T *learning_… in CalSparseApplyProximalAdagrad() argument
89 …size, indices_size, learning_rate, l1_regularization, l2_regularization, gradient, indices, variab… in CalSparseApplyProximalAdagrad()
93 template void CalSparseApplyProximalAdagrad<float>(const size_t size, const size_t indices_size,
99 template void CalSparseApplyProximalAdagrad<half>(const size_t size, const size_t indices_size,
Dscatter_functor_impl.cuh30 …catterFunc(enum ScatterFunctorType func_type, const size_t &inner_size, const size_t &indices_size,
Dsparse_apply_proximal_adagrad_impl.cuh22 void CalSparseApplyProximalAdagrad(const size_t size, const size_t indices_size, const T *learning_…
/third_party/mindspore/mindspore/ccsrc/ps/
Dutil.cc108 void Util::ReduceSparseGradient(float *gradients, int *indices, const size_t indices_size, size_t s… in ReduceSparseGradient() argument
111 size_t slice_segment_size = indices_size * segment_size; in ReduceSparseGradient()
113 std::vector<int> workspace_indices(indices_size); in ReduceSparseGradient()
119 {workspace_grad.data(), workspace_indices.data(), indices_size}); in ReduceSparseGradient()
120 mindspore::kernel::SparseGradient<int> input_sparse_grad({gradients, indices, indices_size}); in ReduceSparseGradient()
Doptimizer_info.cc187 size_t indices_size = static_cast<size_t>(indices()->size / sizeof(int)); in ComputeMean() local
190 std::vector<float> new_grad(indices_size * segment_size); in ComputeMean()
191 std::vector<int> new_indices(indices_size); in ComputeMean()
192 …ernel::SparseGradient<int> unique_sparse_grad({new_grad.data(), new_indices.data(), indices_size}); in ComputeMean()
225 for (size_t j = 0; j < indices_size; j++) { in ComputeMean()
231 …Util::ReduceSparseGradient(grad_data, indices_data, indices_size, segment_size, first_dim_size, ou… in ComputeMean()
Dworker.cc789 size_t indices_size = indice_ids.size(); in SparsePartitioner() local
790 if (indices_size > 0) { in SparsePartitioner()
791 size_t partition_segment_size = indices_size * segment_size; in SparsePartitioner()
793 std::vector<int> src_indice_data(indices_size); in SparsePartitioner()
799 std::vector<int> new_indices(indices_size); in SparsePartitioner()
800 …ernel::SparseGradient<int> unique_sparse_grad({new_grad.data(), new_indices.data(), indices_size}); in SparsePartitioner()
801 …Util::ReduceSparseGradient(src_grad_data.data(), src_indice_data.data(), indices_size, segment_siz… in SparsePartitioner()
819 if (indices_size == 0) { in SparsePartitioner()
Dutil.h54 …static void ReduceSparseGradient(float *gradients, int *indices, const size_t indices_size, size_t…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dbias_add_grad_gpu_kenel.h200 size_t indices_size, workspace_size; in InitSizeLists() local
202 …el_node_, cudnnGetReductionIndicesSize(cudnn_handle_, op_desc_, dy_desc_, db_desc_, &indices_size), in InitSizeLists()
207 workspace_size_list_.push_back(indices_size); in InitSizeLists()
/third_party/mindspore/mindspore/ops/composite/multitype_ops/
D_constexpr_utils.py376 def check_indices(indices_size, index): argument
378 if indices_size < 1:
381 return indices_size
385 def check_indices_value_size(indices_size, value_size): argument
390 if value_size != indices_size:
393 " value size:{}, indics size:{}".format(value_size, indices_size))
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_optimizer_cpu_kernel.h185 size_t indices_size = thread_indices_size; in SplitAndCalculateSegmentBucketSize() local
187 indices_size += 1; in SplitAndCalculateSegmentBucketSize()
192 segments[i]->indices_size_ = indices_size; in SplitAndCalculateSegmentBucketSize()
198 current_indices_offset += indices_size; in SplitAndCalculateSegmentBucketSize()
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py277 indices_size = size_op(new_indices)
278 if indices_size > 0:
280 new_indices_shape_changed = (indices_size,)
490 indices_size = (size_op(indices),)
495 values_shape = indices_size + x_tail_shp
497 indices_new = reshape(indices, indices_size)
/third_party/mindspore/tests/ut/python/ir/
Dtest_row_tensor.py135 indices_size = (size_op(indices),)
137 values_shape = indices_size + x_tail_shp
139 indices = reshape(indices, indices_size)
/third_party/mesa3d/src/mesa/main/
Dglthread_draw.c951 int indices_size = sizeof(indices[0]) * draw_count; in multi_draw_elements_async() local
955 count_size + indices_size + basevertex_size + buffers_size; in multi_draw_elements_async()
969 memcpy(variable_data, indices, indices_size); in multi_draw_elements_async()
970 variable_data += indices_size; in multi_draw_elements_async()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dtensor.cc1142 dsize_t indices_size = static_cast<dsize_t>(indices.size()); in SliceNumeric() local
1143 for (dsize_t i = 0; i < indices_size; i++) { in SliceNumeric()
1145 if (i < indices_size - 1) { in SliceNumeric()
1163 if (i < indices_size - 1) { in SliceNumeric()