Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h85 #define DECONV_WINOGRAD_BUFFER_COUNT 8 macro
119 DeConvWgABuffer a_buffer_[DECONV_WINOGRAD_BUFFER_COUNT];
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc51 for (int i = 0; i < DECONV_WINOGRAD_BUFFER_COUNT; i++) { in FreeResizeBuf()
136 if (unit.winograd_.kh_ >= DECONV_WINOGRAD_BUFFER_COUNT) { in InitParameter()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc48 for (int i = 0; i < DECONV_WINOGRAD_BUFFER_COUNT; i++) { in FreeResizeBuf()
154 if (unit.winograd_.kh_ >= DECONV_WINOGRAD_BUFFER_COUNT) { in InitParameter()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_winograd_fp32.c553 bool transferred[DECONV_WINOGRAD_BUFFER_COUNT] = {false}; in DeconvWg()
561 if (unit->winograd_.kh_ >= DECONV_WINOGRAD_BUFFER_COUNT) { in DeconvWg()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c377 bool transferred[DECONV_WINOGRAD_BUFFER_COUNT] = {false}; in DeconvWgFp16()