Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h99 typedef struct DeConvWgABuffer { struct
103 } DeConvWgABuffer; argument
119 DeConvWgABuffer a_buffer_[DECONV_WINOGRAD_BUFFER_COUNT];
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc49 DeConvWgABuffer &wg = deconv_param_->a_buffer_[i]; in FreeResizeBuf()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc52 DeConvWgABuffer &wg = deconv_param_->a_buffer_[i]; in FreeResizeBuf()
/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()