Home
last modified time | relevance | path

Searched defs:InitWorkspaceSize (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_apply_ftrl_cpu_kernel.cc68 void SparseApplyFtrlCPUKernel::InitWorkspaceSize() { in InitWorkspaceSize() function in mindspore::kernel::SparseApplyFtrlCPUKernel
Dsparse_apply_proximal_adagrad_cpu_kernel.cc63 void SparseApplyProximalAdagradCPUKernel::InitWorkspaceSize() { in InitWorkspaceSize() function in mindspore::kernel::SparseApplyProximalAdagradCPUKernel
Dsparse_apply_lazy_adam_cpu_kernel.cc63 void SparseApplyLazyAdamCPUKernel::InitWorkspaceSize() { in InitWorkspaceSize() function in mindspore::kernel::SparseApplyLazyAdamCPUKernel
Dsparse_apply_adam_cpu_kernel.cc85 void SparseApplyAdamCPUKernel::InitWorkspaceSize() { in InitWorkspaceSize() function in mindspore::kernel::SparseApplyAdamCPUKernel
Dmirror_pad_grad_cpu_kernel.cc143 void MirrorPadGradCPUKernel::InitWorkspaceSize() { in InitWorkspaceSize() function in mindspore::kernel::MirrorPadGradCPUKernel
Dcumsum_cpu_kernel.cc49 void CumSumCPUKernel::InitWorkspaceSize() { in InitWorkspaceSize() function in mindspore::kernel::CumSumCPUKernel