Home
last modified time | relevance | path

Searched defs:all_elements (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Drmsprop_impl.cu74 auto all_elements = batch_size * input_elements; in RmsPropKernel() local
91 auto all_elements = batch_size * input_elements; in RmsPropKernel() local
106 auto all_elements = batch_size * input_elements; in RmsPropKernel() local
130 auto all_elements = batch_size * input_elements; in RmsPropCenterKernel() local
149 auto all_elements = batch_size * input_elements; in RmsPropCenterKernel() local
168 auto all_elements = batch_size * input_elements; in RmsPropCenterKernel() local
189 auto all_elements = batch_size * input_elements; in RmsPropCenterKernel() local
208 auto all_elements = batch_size * input_elements; in RmsPropCenterKernel() local
Dapply_proximal_adagrad_impl.cu59 auto all_elements = input_elements * batch_size; in CalApplyProximalAdagradKernel() local
Dapply_adam_with_amsgrad_impl.cu46 auto all_elements = size * batch_size; in CalApplyAdamWithAmsgradKernel() local
Dapply_adam_with_amsgrad_v2_impl.cu46 auto all_elements = size * batch_size; in CalApplyAdamWithAmsgradV2Kernel() local
Dadam_impl.cu34 auto all_elements = size * batch_size; in ApplyAdamKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Ddevice_matrix.h75 std::vector<std::vector<int64_t>> all_elements = {_s}; in GetAllElements() local
121 std::vector<std::vector<int64_t>> all_elements; in GetAllElements() local