Searched refs:workspace_grad (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ |
| D | sparse_optimizer_cpu_kernel_test.cc | 55 SparseGradient<int> workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F() local
60 param.workspace_grad_ = &workspace_grad; in TEST_F()
107 SparseGradient<int> workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F() local
112 param.workspace_grad_ = &workspace_grad; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/ |
| D | common_utils_test.cc | 55 SparseGradient workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F() local
60 param.workspace_grad_ = &workspace_grad; in TEST_F()
107 SparseGradient workspace_grad({tmp_grad.data(), tmp_indices.data(), 6}); in TEST_F() local
112 param.workspace_grad_ = &workspace_grad; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 138 auto workspace_grad = reinterpret_cast<float *>(workspace[2]->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 | 141 auto *workspace_grad = reinterpret_cast<float *>(workspace[2]->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 | 149 auto *workspace_grad = reinterpret_cast<float *>(workspace[2]->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 | 161 auto *workspace_grad = reinterpret_cast<float *>(workspace[2]->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 | 112 std::vector<float> workspace_grad(slice_segment_size); in ReduceSparseGradient() local
119 {workspace_grad.data(), workspace_indices.data(), indices_size}); in ReduceSparseGradient()
|