| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.h | 39 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/ |
| D | recv_gpu_kernel.h | 33 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | send_gpu_kernel.h | 33 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | rt_kernel.cc | 49 const std::vector<size_t> &RtKernel::GetWorkspaceSizeList() const { return workspace_size_list_; } in GetWorkspaceSizeList() function in mindspore::kernel::RtKernel
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | strided_slice_gpu_kernel.h | 37 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | strided_slice_grad_gpu_kernel.h | 37 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | zeroslike_gpu_kernel.h | 34 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/ |
| D | md_iteration_gradient_descent_kernel.h | 49 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | md_iteration_setup_random_state.h | 50 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | md_iteration_leap_frog_kernel.h | 52 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | md_iteration_leap_frog_with_max_vel_kernel.h | 50 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | logical_not_gpu_kernel.h | 38 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | l2_loss_gpu_kernel.h | 34 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | gelu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | softplus_gpu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | softplus_grad_gpu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | gelu_grad_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | smooth_l1_loss_gpu_kernel.h | 34 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | flatten_gpu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | hsigmoid_gpu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/pme/ |
| D | ifft_3d_kernel.h | 52 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | fft_3d_kernel.h | 52 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
| D | nccl_send_gpu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/random/ |
| D | randperm_gpu_kernel.h | 39 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | fake_learned_scale_quant_perchannel_gpu_kernel.cc | 43 const std::vector<size_t> &FakeLearnedScaleQuantPerChannelGpuKernel::GetWorkspaceSizeList() const { in GetWorkspaceSizeList() function in mindspore::kernel::FakeLearnedScaleQuantPerChannelGpuKernel
|