Searched refs:CreateKernelAddress (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ |
| D | adam_delta_cpu_kernel_test.cc | 41 AddressPtr CreateKernelAddress(void *addr, size_t elem_num) { in CreateKernelAddress() function in mindspore::kernel::AdamDeltaCpuKernelTest
49 inputs_.push_back(CreateKernelAddress(m_.data(), elem_num_)); in CreateAddress()
50 inputs_.push_back(CreateKernelAddress(v_.data(), elem_num_)); in CreateAddress()
51 inputs_.push_back(CreateKernelAddress(&beta1_power_, 1)); in CreateAddress()
52 inputs_.push_back(CreateKernelAddress(&beta2_power_, 1)); in CreateAddress()
53 inputs_.push_back(CreateKernelAddress(&lr_, 1)); in CreateAddress()
54 inputs_.push_back(CreateKernelAddress(&beta1_, 1)); in CreateAddress()
55 inputs_.push_back(CreateKernelAddress(&beta2_, 1)); in CreateAddress()
56 inputs_.push_back(CreateKernelAddress(&epsilon_, 1)); in CreateAddress()
57 inputs_.push_back(CreateKernelAddress(grad_.data(), elem_num_)); in CreateAddress()
[all …]
|
| D | sparse_apply_adam_cpu_kernel_test.cc | 41 AddressPtr CreateKernelAddress(void *addr) { in CreateKernelAddress() function in mindspore::kernel::SparseApplyAdamCpuKernelTest
48 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
49 inputs_.push_back(CreateKernelAddress(m_.data())); in CreateInputAddress()
50 inputs_.push_back(CreateKernelAddress(v_.data())); in CreateInputAddress()
51 inputs_.push_back(CreateKernelAddress(&beta1_power_)); in CreateInputAddress()
52 inputs_.push_back(CreateKernelAddress(&beta2_power_)); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(&lr_)); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(&beta1_)); in CreateInputAddress()
55 inputs_.push_back(CreateKernelAddress(&beta2_)); in CreateInputAddress()
56 inputs_.push_back(CreateKernelAddress(&epsilon_)); in CreateInputAddress()
[all …]
|
| D | sparse_apply_lazy_adam_cpu_kernel_test.cc | 41 AddressPtr CreateKernelAddress(void *addr) { in CreateKernelAddress() function in mindspore::kernel::SparseApplyLazyAdamCpuKernelTest
48 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
49 inputs_.push_back(CreateKernelAddress(m_.data())); in CreateInputAddress()
50 inputs_.push_back(CreateKernelAddress(v_.data())); in CreateInputAddress()
51 inputs_.push_back(CreateKernelAddress(&beta1_power_)); in CreateInputAddress()
52 inputs_.push_back(CreateKernelAddress(&beta2_power_)); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(&lr_)); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(&beta1_)); in CreateInputAddress()
55 inputs_.push_back(CreateKernelAddress(&beta2_)); in CreateInputAddress()
56 inputs_.push_back(CreateKernelAddress(&epsilon_)); in CreateInputAddress()
[all …]
|
| D | unique_with_pad_cpu_kernel_test.cc | 39 AddressPtr CreateKernelAddress(void *addr) { in CreateKernelAddress() function in mindspore::kernel::UniqueWithPadCpuKernelTest
46 inputs_.push_back(CreateKernelAddress(x_.data())); in CreateAddress()
47 inputs_.push_back(CreateKernelAddress(&pad_dim_)); in CreateAddress()
48 outputs_.push_back(CreateKernelAddress(out_.data())); in CreateAddress()
49 outputs_.push_back(CreateKernelAddress(idx_.data())); in CreateAddress()
50 workspace_.push_back(CreateKernelAddress(workspace_idx_.data())); in CreateAddress()
51 workspace_.push_back(CreateKernelAddress(workspace_idx_.data())); in CreateAddress()
52 workspace_.push_back(CreateKernelAddress(workspace_idx_.data())); in CreateAddress()
|
| D | sparse_apply_proximal_adagrad_cpu_kernel_test.cc | 41 AddressPtr CreateKernelAddress(void *addr) { in CreateKernelAddress() function in mindspore::kernel::SparseApplyProximalAdagradCpuKernelTest
48 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
49 inputs_.push_back(CreateKernelAddress(accum_.data())); in CreateInputAddress()
50 inputs_.push_back(CreateKernelAddress(&lr_)); in CreateInputAddress()
51 inputs_.push_back(CreateKernelAddress(&l1_)); in CreateInputAddress()
52 inputs_.push_back(CreateKernelAddress(&l2_)); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(indices.data())); in CreateInputAddress()
59 workspace_.push_back(CreateKernelAddress(new_grad.data())); in CreateWorkspaceAddress()
60 workspace_.push_back(CreateKernelAddress(new_indices.data())); in CreateWorkspaceAddress()
[all …]
|
| D | unique_cpu_kernel_test.cc | 39 AddressPtr CreateKernelAddress(void *addr) { in CreateKernelAddress() function in mindspore::kernel::UniqueCpuKernelTest
46 inputs_.push_back(CreateKernelAddress(x_.data())); in CreateAddress()
47 outputs_.push_back(CreateKernelAddress(y_.data())); in CreateAddress()
48 outputs_.push_back(CreateKernelAddress(idx_.data())); in CreateAddress()
49 workspace_.push_back(CreateKernelAddress(workspace_idx_.data())); in CreateAddress()
50 workspace_.push_back(CreateKernelAddress(workspace_idx_.data())); in CreateAddress()
51 workspace_.push_back(CreateKernelAddress(workspace_idx_.data())); in CreateAddress()
|
| D | sparse_apply_ftrl_cpu_kernel_test.cc | 45 AddressPtr CreateKernelAddress(void *addr) { in CreateKernelAddress() function in mindspore::kernel::SparseApplyFtrlCpuKernelTest
52 inputs_.push_back(CreateKernelAddress(var_.data())); in CreateInputAddress()
53 inputs_.push_back(CreateKernelAddress(accum_.data())); in CreateInputAddress()
54 inputs_.push_back(CreateKernelAddress(linear_.data())); in CreateInputAddress()
55 inputs_.push_back(CreateKernelAddress(grad_.data())); in CreateInputAddress()
56 inputs_.push_back(CreateKernelAddress(indices.data())); in CreateInputAddress()
61 workspace_.push_back(CreateKernelAddress(new_grad.data())); in CreateWorkspaceAddress()
62 workspace_.push_back(CreateKernelAddress(new_indices.data())); in CreateWorkspaceAddress()
63 workspace_.push_back(CreateKernelAddress(tmp_grad.data())); in CreateWorkspaceAddress()
64 workspace_.push_back(CreateKernelAddress(tmp_indices.data())); in CreateWorkspaceAddress()
|