Home
last modified time | relevance | path

Searched refs:var_outer_dim_size (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_apply_proximal_adagrad_cpu_kernel.cc37 const auto var_outer_dim_size = input_params->var_outer_dim_size_; in ComputeProximalAdagrad() local
43 size_t start_index = var_outer_dim_size * static_cast<size_t>(index); in ComputeProximalAdagrad()
44 size_t end_index = start_index + var_outer_dim_size; in ComputeProximalAdagrad()
45 for (size_t j = start_index, k = var_outer_dim_size * i; j < end_index; ++j, ++k) { in ComputeProximalAdagrad()
Dsparse_apply_lazy_adam_cpu_kernel.cc40 const auto var_outer_dim_size = input_params->var_outer_dim_size_; in ComputeLazyAdam() local
46 size_t start_index = var_outer_dim_size * static_cast<size_t>(index); in ComputeLazyAdam()
47 size_t end_index = start_index + var_outer_dim_size; in ComputeLazyAdam()
48 for (size_t j = start_index, k = var_outer_dim_size * i; j < end_index; ++j, ++k) { in ComputeLazyAdam()
Dsparse_apply_ftrl_cpu_kernel.cc39 const auto var_outer_dim_size = input_params->var_outer_dim_size_; in ComputeFtrl() local
45 size_t start_index = var_outer_dim_size * static_cast<size_t>(index); in ComputeFtrl()
46 size_t end_index = start_index + var_outer_dim_size; in ComputeFtrl()
47 for (size_t j = start_index, k = var_outer_dim_size * i; j < end_index; ++j, ++k) { in ComputeFtrl()
Dsparse_apply_adam_cpu_kernel.cc38 const auto var_outer_dim_size = input_params->var_outer_dim_size_; in ComputeAdam() local
44 size_t start_index = var_outer_dim_size * static_cast<size_t>(index); in ComputeAdam()
45 size_t end_index = start_index + var_outer_dim_size; in ComputeAdam()
46 for (size_t j = start_index, k = var_outer_dim_size * i; j < end_index; ++j, ++k) { in ComputeAdam()