Searched refs:middle_buffer_ (Results 1 – 5 of 5) sorted by relevance
55 if (wg.middle_buffer_ != nullptr) { in FreeResizeBuf()56 free(wg.middle_buffer_); in FreeResizeBuf()57 wg.middle_buffer_ = nullptr; in FreeResizeBuf()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()417 wg.middle_buffer_ = nullptr; in Init()
58 if (wg.middle_buffer_ != nullptr) { in FreeResizeBuf()59 free(wg.middle_buffer_); in FreeResizeBuf()60 wg.middle_buffer_ = nullptr; in FreeResizeBuf()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()
101 void *middle_buffer_; member
565 …float *wg_mid_a_buf = (float *)wg_buf->middle_buffer_ + task_id * unit->winograd_.kw_ * unit->wino… in DeconvWg()
386 …float16_t *mid_a = (float16_t *)tmp_a->middle_buffer_ + task_id * unit->winograd_.kw_ * unit->wino… in DeconvWgFp16()