Home
last modified time | relevance | path

Searched defs:indices_size (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dsparse_apply_proximal_adagrad_impl.cu60 __global__ void SparseApplyProximalAdagradUpdate(const size_t size, const size_t indices_size, cons… in SparseApplyProximalAdagradUpdate()
84 void CalSparseApplyProximalAdagrad(const size_t size, const size_t indices_size, const T *learning_… in CalSparseApplyProximalAdagrad()
Dscatter_functor_impl.cu54 …catterFunc(enum ScatterFunctorType func_type, const size_t &inner_size, const size_t &indices_size, in ScatterFunc()
/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
/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()
Doptimizer_info.cc187 size_t indices_size = static_cast<size_t>(indices()->size / sizeof(int)); in ComputeMean() local
Dworker.cc789 size_t indices_size = indice_ids.size(); in SparsePartitioner() local
/third_party/mindspore/mindspore/ops/composite/multitype_ops/
D_constexpr_utils.py376 def check_indices(indices_size, index): argument
385 def check_indices_value_size(indices_size, value_size): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_optimizer_cpu_kernel.h185 size_t indices_size = thread_indices_size; in SplitAndCalculateSegmentBucketSize() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dtensor.cc1142 dsize_t indices_size = static_cast<dsize_t>(indices.size()); in SliceNumeric() local
/third_party/mesa3d/src/mesa/main/
Dglthread_draw.c951 int indices_size = sizeof(indices[0]) * draw_count; in multi_draw_elements_async() local