| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | apply_momentum_ps_kernel.cc | 27 const std::vector<size_t> &ApplyMomentumPSKernel::input_sizes() const { return GetInputSizeList(); } in input_sizes()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | fake_learned_scale_quant_perlayer_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | fake_learned_scale_quant_perlayer_grad_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | fake_learned_scale_quant_perchannel_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | minmax_update_perlayer_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | minmax_update_perchannel_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | fake_quant_perchannel_grad_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | fake_learned_scale_quant_perchannel_grad_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | fake_quant_perlayer_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | fake_quant_perlayer_grad_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | fake_quant_perchannel_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | rt_kernel.cc | 45 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/rl/ |
| D | buffer_append_gpu_kernel.h | 33 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | buffer_get_gpu_kernel.h | 33 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | buffer_sample_gpu_kernel.h | 35 const std::vector<size_t> &GetInputSizeList() const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/ |
| D | dataset_init_kernel.h | 32 const std::vector<size_t> &GetInputSizeList() const override;
|
| D | dataset_iterator_kernel.h | 34 const std::vector<size_t> &GetInputSizeList() const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/trt/ |
| D | trt_kernel.h | 35 const std::vector<size_t> &GetInputSizeList() const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/control/ |
| D | recv_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
|
| D | send_gpu_kernel.h | 31 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_mod.h | 35 const std::vector<size_t> &GetInputSizeList() const override;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/ |
| D | md_iteration_gradient_descent_kernel.h | 47 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
|
| D | md_iteration_setup_random_state.h | 48 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/ |
| D | gelu_kernel.h | 33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
|
| D | softplus_gpu_kernel.h | 33 const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; } in GetInputSizeList() function
|