Searched refs:input_sparse_grad (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 143 SparseGradient<T> input_sparse_grad({grad, indices, indices_size_}); in LaunchKernel() local
145 param.input_grad_ = &input_sparse_grad; in LaunchKernel()
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 146 SparseGradient<T> input_sparse_grad({grad, indices, indices_size_}); in LaunchKernel() local
148 param.input_grad_ = &input_sparse_grad; in LaunchKernel()
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 154 SparseGradient<T> input_sparse_grad({grad, indices, indices_size_}); in LaunchKernel() local
156 param.input_grad_ = &input_sparse_grad; in LaunchKernel()
|
| D | sparse_apply_adam_cpu_kernel.cc | 167 SparseGradient<T> input_sparse_grad({grad, indices, indices_size_}); in LaunchKernel() local
169 param.input_grad_ = &input_sparse_grad; in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.cc | 120 mindspore::kernel::SparseGradient<int> input_sparse_grad({gradients, indices, indices_size}); in ReduceSparseGradient() local
122 param.input_grad_ = &input_sparse_grad; in ReduceSparseGradient()
|