Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc41 if (unit.winograd_.b_buffer_ != nullptr) { in FreeResizeBuf()
42 free(unit.winograd_.b_buffer_); in FreeResizeBuf()
43 unit.winograd_.b_buffer_ = nullptr; in FreeResizeBuf()
173 …unit.winograd_.b_buffer_ = malloc(deconv_param_->thread_num_ * unit.winograd_.kh_ * unit.winograd_… in InitParameter()
175 if (unit.winograd_.b_buffer_ == nullptr) { in InitParameter()
308 unit.winograd_.b_buffer_ = nullptr; in InitComputeParam()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc44 if (unit.winograd_.b_buffer_ != nullptr) { in FreeResizeBuf()
45 free(unit.winograd_.b_buffer_); in FreeResizeBuf()
46 unit.winograd_.b_buffer_ = nullptr; in FreeResizeBuf()
158 …unit.winograd_.b_buffer_ = malloc(deconv_param_->thread_num_ * unit.winograd_.kh_ * unit.winograd_… in InitParameter()
160 if (unit.winograd_.b_buffer_ == nullptr) { in InitParameter()
265 unit.winograd_.b_buffer_ = nullptr; in InitComputeParam()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h87 void *b_buffer_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_winograd_fp32.c569 …float *tmp_b_buf = (float *)unit->winograd_.b_buffer_ + task_id * unit->winograd_.kh_ * unit->wino… in DeconvWg()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c390 …float16_t *tmp_b = (float16_t *)unit->winograd_.b_buffer_ + task_id * unit->winograd_.kh_ * unit->… in DeconvWgFp16()