Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OpenCV
opencv
提交
924a6541
O
opencv
项目概览
OpenCV
/
opencv
上一次同步 大约 1 年
通知
1005
Star
71102
Fork
55580
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
opencv
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
924a6541
编写于
3月 15, 2023
作者:
A
Alexander Smorkalov
提交者:
GitHub
3月 15, 2023
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #23357 from zihaomu:fix_winograd_error_32bit
DNN : fix bug in 32 bit cpu
上级
ccbc7841
6bac5453
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
5 addition
and
5 deletion
+5
-5
modules/dnn/src/layers/cpu_kernels/convolution.cpp
modules/dnn/src/layers/cpu_kernels/convolution.cpp
+3
-3
modules/dnn/src/layers/cpu_kernels/convolution.hpp
modules/dnn/src/layers/cpu_kernels/convolution.hpp
+2
-2
未找到文件。
modules/dnn/src/layers/cpu_kernels/convolution.cpp
浏览文件 @
924a6541
...
...
@@ -103,13 +103,13 @@ Ptr<FastConv> initFastConv(
}
conv
->
conv_type
=
ifRunDepthWise
&&
conv_dim
!=
CONV_3D
?
CONV_TYPE_DEPTHWISE
:
useWinograd
&&
(
conv_dim
==
CONV_2D
&&
(
conv
->
useSIMD128
||
conv
->
useAVX2
||
conv
->
useNEON
)
&&
useWinograd
&&
(
conv_dim
==
CONV_2D
&&
(
conv
->
useSIMD128
||
conv
->
useAVX
||
conv
->
useAVX
2
||
conv
->
useNEON
)
&&
Hk
==
3
&&
Wk
==
3
&&
dilation_h
==
1
&&
dilation_w
==
1
&&
stride_h
==
1
&&
stride_w
==
1
)
?
CONV_TYPE_WINOGRAD3X3
:
(
ifRunDepthWiseRemain
?
CONV_TYPE_DEPTHWISE_REMAIN
:
CONV_TYPE_GENERIC
);
#if !(CV_NEON || CV_SIMD128 || CV_TRY_AVX2)
if
(
conv
->
conv_type
==
CONV_TYPE_WINOGRAD3X3
)
// Disabel Winograd when CV_NEON, CV_SIMD128 and CV_TRY_AVX2 are not available.
#if !(CV_NEON || CV_SIMD128 || CV_TRY_AVX
|| CV_TRY_AVX
2)
if
(
conv
->
conv_type
==
CONV_TYPE_WINOGRAD3X3
)
// Disabel Winograd when CV_NEON, CV_SIMD128
,CV_TRY_AVX
and CV_TRY_AVX2 are not available.
conv
->
conv_type
=
CONV_TYPE_GENERIC
;
#endif
...
...
modules/dnn/src/layers/cpu_kernels/convolution.hpp
浏览文件 @
924a6541
...
...
@@ -28,13 +28,13 @@ enum {
CONV_WINO_AREA
=
CONV_WINO_SIZE
*
CONV_WINO_SIZE
,
CONV_WINO_KBLOCK
=
4
,
#if (CV_NEON && CV_NEON_AARCH64) || CV_TRY_AVX2
#if (CV_NEON && CV_NEON_AARCH64) || CV_TRY_AVX
|| CV_TRY_AVX
2
CONV_WINO_IBLOCK
=
6
,
#else
CONV_WINO_IBLOCK
=
3
,
#endif
#if CV_TRY_AVX2
#if CV_TRY_AVX
|| CV_TRY_AVX
2
CONV_WINO_ATOM_F32
=
8
,
#else
CONV_WINO_ATOM_F32
=
4
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录