| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/kernel/cpu/ |
| D | sparse_apply_proximal_adagrad_cpu_kernel_test.cc | 61 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress()
131 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
161 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
195 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
|
| D | sparse_apply_ftrl_cpu_kernel_test.cc | 62 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress()
133 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
167 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
205 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
|
| D | sparse_apply_lazy_adam_cpu_kernel_test.cc | 64 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress()
144 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
175 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
210 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
|
| D | sparse_apply_adam_cpu_kernel_test.cc | 64 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress()
146 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
178 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
214 std::vector<float> new_grad(3 * 3 * 3); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | fused_sparse_ftrl.cc | 75 float *new_grad; in Compute() local
|
| D | fused_sparse_lazy_adam.cc | 78 float *new_grad; in Compute() local
|
| D | fused_sparse_proximal_adagrad.cc | 81 float *new_grad; in DoCompute() local
|
| D | fused_sparse_adam.cc | 98 float *new_grad; in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_apply_adagrad_cpu_kernel.cc | 211 auto *new_grad = reinterpret_cast<float *>(workspace[0]->device_ptr()); in LaunchKernel() local
|
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 240 auto new_grad = reinterpret_cast<float *>(workspace[kWorkSpaceIndex0]->device_ptr()); in LaunchKernel() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 248 auto *new_grad = reinterpret_cast<float *>(workspace[0]->device_ptr()); in LaunchKernel() local
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 252 auto *new_grad = reinterpret_cast<float *>(workspace[0]->device_ptr()); in LaunchKernel() local
|
| D | sparse_apply_adam_cpu_kernel.cc | 272 auto *new_grad = reinterpret_cast<float *>(workspace[0]->device_ptr()); in LaunchKernel() local
|