| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | apply_momentum_ps_kernel.cc | 31 const std::vector<size_t> &ApplyMomentumPSKernel::workspace_sizes() const { return GetWorkspaceSize… in workspace_sizes() function in mindspore::kernel::ps::ApplyMomentumPSKernel
|
| D | sparse_apply_ftrl_ps_kernel.cc | 126 const std::vector<size_t> &SparseApplyFtrlPSKernel::workspace_sizes() const { return GetWorkspaceSi… in workspace_sizes() function in mindspore::kernel::ps::SparseApplyFtrlPSKernel
|
| D | sparse_apply_adam_ps_kernel.cc | 115 const std::vector<size_t> &SparseApplyAdamPSKernel::workspace_sizes() const { return GetWorkspaceSi… in workspace_sizes() function in mindspore::kernel::ps::SparseApplyAdamPSKernel
|
| D | sparse_apply_lazy_adam_ps_kernel.cc | 117 const std::vector<size_t> &SparseApplyLazyAdamPSKernel::workspace_sizes() const { return GetWorkspa… in workspace_sizes() function in mindspore::kernel::ps::SparseApplyLazyAdamPSKernel
|
| D | embedding_look_up_ps_kernel.cc | 111 const std::vector<size_t> &EmbeddingLookUpPSKernel::workspace_sizes() const { return GetWorkspaceSi… in workspace_sizes() function in mindspore::kernel::ps::EmbeddingLookUpPSKernel
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 219 auto workspace_sizes = op->GetWorkspaceBytes(); in ParseDomiTask() local
|
| D | hccl_adapter.cc | 254 auto workspace_sizes = op->GetWorkspaceBytes(); in CalcWorkspaceSize() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 640 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitKernelWorkspaceAddress() local
705 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in ClearKernelWorkspaceAddress() local
1165 auto workspace_sizes = kernel_mod.GetWorkspaceSizeList(); in AllocKernelWorkspaceDynamicRes() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse.cc | 111 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitDynamicWorkspaceKernelRef() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 168 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in AssignKernelOutputAddress() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 206 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in CreateKernelWorkspaceDeviceAddress() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 466 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in InitSomasOutputAndWorkspaceTensors() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 307 auto workspace_sizes = kernel_mod->GetWorkspaceSizeList(); in ResetNodeAddress() local
|