Home
last modified time | relevance | path

Searched defs:GetOutputSizeList (Results 1 – 25 of 290) sorted by relevance

12345678910>>...12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_mod.h38 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/
Drecv_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dsend_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Drt_kernel.cc47 const std::vector<size_t> &RtKernel::GetOutputSizeList() const { return output_size_list_; } in GetOutputSizeList() function in mindspore::kernel::RtKernel
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dstrided_slice_gpu_kernel.h36 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dstrided_slice_grad_gpu_kernel.h36 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dzeroslike_gpu_kernel.h33 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/
Dmd_iteration_gradient_descent_kernel.h48 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dmd_iteration_setup_random_state.h49 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dmd_iteration_leap_frog_kernel.h51 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dmd_iteration_leap_frog_with_max_vel_kernel.h49 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dlogical_not_gpu_kernel.h37 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dl2_loss_gpu_kernel.h33 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dgelu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dsoftplus_gpu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dsoftplus_grad_gpu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dgelu_grad_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dsmooth_l1_loss_gpu_kernel.h33 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dflatten_gpu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dhsigmoid_gpu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/pme/
Difft_3d_kernel.h51 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dfft_3d_kernel.h51 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_send_gpu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/random/
Drandperm_gpu_kernel.h38 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/
Dfake_learned_scale_quant_perchannel_gpu_kernel.cc39 const std::vector<size_t> &FakeLearnedScaleQuantPerChannelGpuKernel::GetOutputSizeList() const { in GetOutputSizeList() function in mindspore::kernel::FakeLearnedScaleQuantPerChannelGpuKernel

12345678910>>...12