Home
last modified time | relevance | path

Searched defs:GetInputSizeList (Results 1 – 25 of 289) sorted by relevance

12345678910>>...12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.h37 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/
Drecv_gpu_kernel.h31 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dsend_gpu_kernel.h31 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dstrided_slice_gpu_kernel.h35 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dstrided_slice_grad_gpu_kernel.h35 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dzeroslike_gpu_kernel.h32 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Drt_kernel.cc45 const std::vector<size_t> &RtKernel::GetInputSizeList() const { return input_size_list_; } in GetInputSizeList() function in mindspore::kernel::RtKernel
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/
Dmd_iteration_gradient_descent_kernel.h47 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dmd_iteration_setup_random_state.h48 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dmd_iteration_leap_frog_kernel.h50 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dmd_iteration_leap_frog_with_max_vel_kernel.h48 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dgelu_kernel.h33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dl2_loss_gpu_kernel.h32 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dsoftplus_gpu_kernel.h33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dsoftplus_grad_gpu_kernel.h33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dgelu_grad_kernel.h33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dsmooth_l1_loss_gpu_kernel.h32 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dflatten_gpu_kernel.h33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Drelu_grad_v2_gpu_kernel.h35 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Drelu_v2_gpu_kernel.h35 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dlogical_not_gpu_kernel.h36 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/pme/
Difft_3d_kernel.h50 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
Dfft_3d_kernel.h50 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_send_gpu_kernel.h33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/random/
Drandperm_gpu_kernel.h37 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function

12345678910>>...12