Home
last modified time | relevance | path

Searched refs:GetOutputSizeList (Results 1 – 25 of 335) sorted by relevance

12345678910>>...14

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/
Dapply_momentum_ps_kernel.cc29 const std::vector<size_t> &ApplyMomentumPSKernel::output_sizes() const { return GetOutputSizeList()… in output_sizes()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_all_to_all.h32 const std::vector<size_t> &GetOutputSizeList() const override;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/
Dfake_learned_scale_quant_perlayer_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dfake_learned_scale_quant_perlayer_grad_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dminmax_update_perchannel_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dminmax_update_perlayer_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dfake_learned_scale_quant_perchannel_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dfake_quant_perchannel_grad_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dfake_learned_scale_quant_perchannel_grad_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dfake_quant_perlayer_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dfake_quant_perlayer_grad_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
Dfake_quant_perchannel_gpu_kernel.h32 const std::vector<size_t> &GetOutputSizeList() const override;
/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/rl/
Dbuffer_append_gpu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override;
Dbuffer_get_gpu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override;
Dbuffer_sample_gpu_kernel.h36 const std::vector<size_t> &GetOutputSizeList() const override;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/
Ddataset_init_kernel.h33 const std::vector<size_t> &GetOutputSizeList() const override;
Ddataset_iterator_kernel.h35 const std::vector<size_t> &GetOutputSizeList() const override;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/
Dtrt_kernel.h36 const std::vector<size_t> &GetOutputSizeList() const override;
/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/akg/ascend/
Dakg_ascend_kernel_mod.h36 const std::vector<size_t> &GetOutputSizeList() const override;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/
Dmd_iteration_setup_random_state.h49 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
Dmd_iteration_gradient_descent_kernel.h48 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/
Dgelu_kernel.h34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function

12345678910>>...14