Searched refs:workspace_indices (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 | 139 auto workspace_indices = reinterpret_cast<T *>(workspace[3]->addr); in LaunchKernel() local
142 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel()
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 142 auto *workspace_indices = reinterpret_cast<T *>(workspace[3]->addr); in LaunchKernel() local
145 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel()
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 150 auto *workspace_indices = reinterpret_cast<T *>(workspace[3]->addr); in LaunchKernel() local
153 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel()
|
| D | sparse_apply_adam_cpu_kernel.cc | 162 auto *workspace_indices = reinterpret_cast<T *>(workspace[3]->addr); in LaunchKernel() local
166 SparseGradient<T> workspace_sparse_grad({workspace_grad, workspace_indices, indices_size_}); in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.cc | 113 std::vector<int> workspace_indices(indices_size); in ReduceSparseGradient() local
119 {workspace_grad.data(), workspace_indices.data(), indices_size}); in ReduceSparseGradient()
|