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

conv1x1 op

This commit is contained in:
lzk 2021-05-26 02:35:00 -07:00
parent d5f2fb40f1
commit f0ac8f5228
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/src/runtime/kernel/arm/fp32/convolution_depthwise_slidewindow_x86_fp32.cc
View File

@ -166,8 +166,8 @@ int ConvolutionDepthwiseSWCPUKernelX86::Run() {
packed_output_ = output_ptr; packed_output_ = output_ptr;
} }
ret = ParallelLaunch(static_cast<const lite::InnerContext *>(this->context_)->thread_pool_, ConvDwSWAvxRun, this, ret = static_cast<const lite::InnerContext *>(this->context_)
conv_param_->thread_num_); ->thread_pool_->ParallelLaunch(ConvDwSWAvxRun, this, conv_param_->thread_num_);
if (ret != RET_OK) { if (ret != RET_OK) {
MS_LOG(ERROR) << "ConvDwSWAvxRun error: error_code[" << ret << "]"; MS_LOG(ERROR) << "ConvDwSWAvxRun error: error_code[" << ret << "]";
} }