Home
last modified time | relevance | path

Searched refs:use_winograd_ (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_winograd_fp32.cc43 if (unit.use_winograd_) { in FreeResizeBuf()
83 if (unit.use_winograd_) { in FreeDeconvParam()
135 if (unit.use_winograd_) { in InitParameter()
257 unit.use_winograd_ = true; in InitComputeParam()
273 unit.use_winograd_ = false; in InitComputeParam()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc40 if (unit.use_winograd_) { in FreeResizeBuf()
90 if (unit.use_winograd_) { in FreeDeconvParam()
153 if (unit.use_winograd_) { in InitParameter()
300 unit.use_winograd_ = true; in InitComputeParam()
315 unit.use_winograd_ = false; in InitComputeParam()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dwinograd.h30 use_winograd_ = true; in WinogradOpenCLKernel()
Dconv2d.h76 bool use_winograd_{false};
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h112 bool use_winograd_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_winograd_fp32.c75 if (unit->use_winograd_) { in PackDeConvWgDataFp32()
556 if (unit->use_winograd_) { in DeconvWg()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_winograd_fp16.c265 if (unit->use_winograd_) { in PackDeConvWgDataFp16()
380 if (unit->use_winograd_) { in DeconvWgFp16()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/
Dopencl_fusion.cc399 if (reinterpret_cast<Conv2DOpenCLKernel *>(conv->kernel())->use_winograd_) { in TryMergeConvPReLU()