mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

conv common

This commit is contained in:
lzk 2021-07-08 06:45:50 -07:00
parent 8c6168e5a9
commit 9c633d5897
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/runtime/kernel/arm/fp32/convolution_delegate_fp32.cc
View File

@ -168,7 +168,7 @@ kernel::InnerKernel *ConvolutionDelegateCPUKernel::CpuConvFp32KernelSelect() {
} else {
#ifdef ENABLE_AVX
if (conv_param->input_channel_ / op_parameter_->thread_num_ > 64 ||
conv_param->input_h_ < conv_param->thread_num_) {
conv_param->input_h_ < conv_param->thread_num_ || conv_param->kernel_h_ >= 7 || conv_param->kernel_w_ >= 7) {
kernel = new (std::nothrow) kernel::ConvolutionCPUKernel(
op_parameter_, in_tensors_, out_tensors_, static_cast<const lite::InnerContext *>(this->context_),
origin_weight_, origin_bias_);