Home
last modified time | relevance | path

Searched defs:CreateWorkspaceAddress (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/kernel/cpu/
Dsparse_apply_proximal_adagrad_cpu_kernel_test.cc61 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::SparseApplyProximalAdagradCpuKernelTest
Dsparse_apply_ftrl_cpu_kernel_test.cc62 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::FusedSparseFtrlCpuKernelTest
Dsparse_apply_lazy_adam_cpu_kernel_test.cc64 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::SparseApplyLazyAdamCpuKernelTest
Dsparse_apply_adam_cpu_kernel_test.cc64 void CreateWorkspaceAddress(std::vector<float> &new_grad, std::vector<int64_t> &new_indices, in CreateWorkspaceAddress() function in mindspore::kernel::SparseApplyAdamCpuKernelTest
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Ddevice_address_utils.cc1249 device::DeviceAddressPtr DeviceAddressUtils::CreateWorkspaceAddress(const DeviceContext *device_con… in CreateWorkspaceAddress() function in mindspore::runtime::DeviceAddressUtils