Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc49 DeConvWgABuffer &wg = deconv_param_->a_buffer_[i]; in FreeResizeBuf()
157 if (deconv_param_->a_buffer_[unit.winograd_.kh_].buf_init_ == false) { in InitParameter()
158 deconv_param_->a_buffer_[unit.winograd_.kh_].buf_init_ = true; in InitParameter()
161 deconv_param_->a_buffer_[unit.winograd_.kh_].middle_buffer_ = in InitParameter()
163 if (deconv_param_->a_buffer_[unit.winograd_.kh_].middle_buffer_ == nullptr) { in InitParameter()
166 deconv_param_->a_buffer_[unit.winograd_.kh_].dest_buffer_ = in InitParameter()
168 if (deconv_param_->a_buffer_[unit.winograd_.kh_].dest_buffer_ == nullptr) { in InitParameter()
414 for (auto &wg : deconv_param_->a_buffer_) { in Init()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc52 DeConvWgABuffer &wg = deconv_param_->a_buffer_[i]; in FreeResizeBuf()
140 if (deconv_param_->a_buffer_[unit.winograd_.kh_].buf_init_ == false) { in InitParameter()
141 deconv_param_->a_buffer_[unit.winograd_.kh_].buf_init_ = true; in InitParameter()
144 deconv_param_->a_buffer_[unit.winograd_.kh_].middle_buffer_ = in InitParameter()
146 if (deconv_param_->a_buffer_[unit.winograd_.kh_].middle_buffer_ == nullptr) { in InitParameter()
150 deconv_param_->a_buffer_[unit.winograd_.kh_].dest_buffer_ = in InitParameter()
152 if (deconv_param_->a_buffer_[unit.winograd_.kh_].dest_buffer_ == nullptr) { in InitParameter()
370 for (auto &wg : deconv_param_->a_buffer_) { in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h119 DeConvWgABuffer a_buffer_[DECONV_WINOGRAD_BUFFER_COUNT]; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_winograd_fp32.c564 DeConvWgABuffer *wg_buf = &deconv_param->a_buffer_[unit->winograd_.kh_]; in DeconvWg()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c385 DeConvWgABuffer *tmp_a = &deconv_param->a_buffer_[unit->winograd_.kh_]; in DeconvWgFp16()