Home
last modified time | relevance | path

Searched refs:compute_size_ (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc34 for (int i = 0; i < deconv_param_->compute_size_; i++) { in FreeResizeBuf()
82 for (int i = 0; i < deconv_param_->compute_size_; i++) { in FreeDeconvParam()
151 for (int i = 0; i < deconv_param_->compute_size_; i++) { in InitParameter()
265 deconv_param_->compute_size_ = 0; in InitComputeParam()
269 deconv_param_->compute_size_++; in InitComputeParam()
274 int size = deconv_param_->compute_size_ * sizeof(DeConvComputeUnit); in InitComputeParam()
339 for (int i = 0; i < deconv_param_->compute_size_; i++) { in InitDataParam()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc36 for (int i = 0; i < deconv_param_->compute_size_; i++) { in FreeResizeBuf()
75 for (int i = 0; i < deconv_param_->compute_size_; i++) { in FreeDeconvParam()
133 for (int i = 0; i < deconv_param_->compute_size_; i++) { in InitParameter()
218 deconv_param_->compute_size_ = 0; in InitComputeParam()
222 deconv_param_->compute_size_++; in InitComputeParam()
227 int size = deconv_param_->compute_size_ * sizeof(DeConvComputeUnit); in InitComputeParam()
299 for (int i = 0; i < deconv_param_->compute_size_; i++) { in InitDataParam()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h118 int compute_size_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_winograd_fp32.c554 for (int i = 0; i < deconv_param->compute_size_; i++) { in DeconvWg()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c378 for (int i = 0; i < deconv_param->compute_size_; i++) { in DeconvWgFp16()