| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/ |
| D | send_gpu_kernel.h | 33 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | recv_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/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/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/nn/ |
| D | gelu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | l2_loss_gpu_kernel.h | 34 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 | fused_add_relu_grad_v2_gpu_kernel.h | 37 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
|
| D | assign_add_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/nccl/ |
| D | nccl_send_gpu_kernel.h | 35 const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; } in GetWorkspaceSizeList() function
|
| D | nccl_recv_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
|