| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ |
| D | sparse_apply_ftrl_cpu_kernel_test.cc | 59 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::SparseApplyFtrlCpuKernelTest
95 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
122 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
153 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
|
| D | sparse_apply_proximal_adagrad_cpu_kernel_test.cc | 57 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::SparseApplyProximalAdagradCpuKernelTest
94 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
120 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
150 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
|
| D | sparse_apply_lazy_adam_cpu_kernel_test.cc | 61 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::SparseApplyLazyAdamCpuKernelTest
103 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
130 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
161 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices); in TEST_F()
|
| D | sparse_apply_adam_cpu_kernel_test.cc | 61 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::SparseApplyAdamCpuKernelTest
106 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
134 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
166 CreateWorkspaceAddress(new_grad, new_indices, tmp_grad, tmp_indices, m_t); in TEST_F()
|