| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | apply_momentum_ps_kernel.cc | 29 const std::vector<size_t> &ApplyMomentumPSKernel::output_sizes() const { return GetOutputSizeList()… in output_sizes() function in mindspore::kernel::ps::ApplyMomentumPSKernel
|
| D | sparse_apply_adam_ps_kernel.cc | 113 const std::vector<size_t> &SparseApplyAdamPSKernel::output_sizes() const { return GetOutputSizeList… in output_sizes() function in mindspore::kernel::ps::SparseApplyAdamPSKernel
|
| D | sparse_apply_ftrl_ps_kernel.cc | 124 const std::vector<size_t> &SparseApplyFtrlPSKernel::output_sizes() const { return GetOutputSizeList… in output_sizes() function in mindspore::kernel::ps::SparseApplyFtrlPSKernel
|
| D | sparse_apply_lazy_adam_ps_kernel.cc | 115 const std::vector<size_t> &SparseApplyLazyAdamPSKernel::output_sizes() const { return GetOutputSize… in output_sizes() function in mindspore::kernel::ps::SparseApplyLazyAdamPSKernel
|
| D | embedding_look_up_ps_kernel.cc | 109 const std::vector<size_t> &EmbeddingLookUpPSKernel::output_sizes() const { return GetOutputSizeList… in output_sizes() function in mindspore::kernel::ps::EmbeddingLookUpPSKernel
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 621 auto output_sizes = kernel_mod->GetOutputSizeList(); in InitKernelOutputAddress() local
683 auto output_sizes = kernel_mod->GetOutputSizeList(); in ClearKernelOutputAddress() local
1144 auto output_sizes = kernel_mod.GetOutputSizeList(); in AllocKernelOutputDynamicRes() local
1244 auto output_sizes = kernel_mod->GetOutputSizeList(); in AllocCommunicationOpOutputDynamicRes() local
1374 auto output_sizes = kernel_mod->GetOutputSizeList(); in GetMutableOutputAddr() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 300 auto output_sizes = kernel_mod->GetOutputSizeList(); in ResetNodeAddress() local
442 auto output_sizes = kernel_mod->GetOutputSizeList(); in RunOpAssignOutputMemory() local
753 auto output_sizes = kernel_mod->GetOutputSizeList(); in AssignCommunicationNodeOutputMem() local
825 auto output_sizes = kernel_mod->GetOutputSizeList(); in PreAssignCNodeMemory() local
905 auto output_sizes = kernel_mod->GetOutputSizeList(); in AssignNodeOutputMem() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse.cc | 49 auto output_sizes = kernel_mod->GetOutputSizeList(); in InitDynamicOutputKernelRef() local
|
| D | mem_reuse_checker.cc | 225 auto output_sizes = kernel_mod->GetOutputSizeList(); in CheckGraphOutputAssigned() local
|
| D | mem_swap_manager.cc | 38 auto output_sizes = kernel_mod->GetOutputSizeList(); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | data_prepare_actor.cc | 82 const auto &output_sizes = kernel_mod->GetOutputSizeList(); in FetchContinuousMemoryInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 161 auto output_sizes = kernel_mod->GetOutputSizeList(); in AssignKernelOutputAddress() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 181 auto output_sizes = kernel_mod->GetOutputSizeList(); in CreateKernelOutputDeviceAddress() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 439 auto output_sizes = kernel_mod->GetOutputSizeList(); in InitSomasOutputAndWorkspaceTensors() local
801 auto output_sizes = kernel_mod->GetOutputSizeList(); in RefNodeProcess() local
|