Searched defs:unique_sparse_grad (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_ftrl_cpu_kernel.cc | 37 const auto unique_sparse_grad = input_params->sparse_grad_; in ComputeFtrl() local
152 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel() local
|
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 35 const auto unique_sparse_grad = input_params->sparse_grad_; in ComputeProximalAdagrad() local
141 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 38 const auto unique_sparse_grad = input_params->sparse_grad_; in ComputeLazyAdam() local
144 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel() local
|
| D | sparse_apply_adam_cpu_kernel.cc | 36 const auto unique_sparse_grad = input_params->sparse_grad_; in ComputeAdam() local
165 SparseGradient<T> unique_sparse_grad({new_grad, new_indices, indices_size_}); in LaunchKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.cc | 110 mindspore::kernel::SparseGradient<int> *unique_sparse_grad) { in ReduceSparseGradient()
|
| D | optimizer_info.cc | 192 …mindspore::kernel::SparseGradient<int> unique_sparse_grad({new_grad.data(), new_indices.data(), in… in ComputeMean() local
|
| D | worker.cc | 800 …mindspore::kernel::SparseGradient<int> unique_sparse_grad({new_grad.data(), new_indices.data(), in… in SparsePartitioner() local
|