/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
D | apply_momentum_ps_kernel.cc | 29 const std::vector<size_t> &ApplyMomentumPSKernel::output_sizes() const { return GetOutputSizeList()… in output_sizes()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
D | hcom_all_to_all.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
D | fake_learned_scale_quant_perlayer_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | fake_learned_scale_quant_perlayer_grad_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | minmax_update_perchannel_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | minmax_update_perlayer_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | fake_learned_scale_quant_perchannel_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | fake_quant_perchannel_grad_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | fake_learned_scale_quant_perchannel_grad_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | fake_quant_perlayer_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | fake_quant_perlayer_grad_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | fake_quant_perchannel_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override;
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
D | rt_kernel.cc | 47 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/ |
D | buffer_append_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | buffer_get_gpu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | buffer_sample_gpu_kernel.h | 36 const std::vector<size_t> &GetOutputSizeList() const override;
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/ |
D | dataset_init_kernel.h | 33 const std::vector<size_t> &GetOutputSizeList() const override;
|
D | dataset_iterator_kernel.h | 35 const std::vector<size_t> &GetOutputSizeList() const override;
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/ |
D | trt_kernel.h | 36 const std::vector<size_t> &GetOutputSizeList() const override;
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/ |
D | recv_gpu_kernel.h | 32 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
|
D | send_gpu_kernel.h | 32 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/ |
D | akg_ascend_kernel_mod.h | 36 const std::vector<size_t> &GetOutputSizeList() const override;
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/ |
D | md_iteration_setup_random_state.h | 49 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
|
D | md_iteration_gradient_descent_kernel.h | 48 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/ |
D | gelu_kernel.h | 34 const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; } in GetOutputSizeList() function
|