Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[v1.5.x]Add MKLDNN 4c layout to fix gluoncv se_resnext101_64x4d (#15692) #15801

Merged
merged 1 commit into from
Aug 8, 2019
Merged

[v1.5.x]Add MKLDNN 4c layout to fix gluoncv se_resnext101_64x4d (#15692) #15801

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/operator/nn/mkldnn/mkldnn_base.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -329,6 +329,7 @@ mkldnn_memory_format_t GetDefaultFormat(const mkldnn::memory::desc &desc) {
case mkldnn_nchw:
case mkldnn_nhwc:
case mkldnn_chwn:
case mkldnn_nChw4c:
case mkldnn_nChw8c:
case mkldnn_nChw16c:
return mkldnn_nchw;
Expand All @@ -338,6 +339,7 @@ mkldnn_memory_format_t GetDefaultFormat(const mkldnn::memory::desc &desc) {
case mkldnn_iohw:
case mkldnn_oIhw8i:
case mkldnn_oIhw16i:
case mkldnn_OIhw4i4o:
case mkldnn_OIhw8i8o:
case mkldnn_hwio_s8s8:
case mkldnn_OIhw16i16o:
Expand Down Expand Up @@ -376,13 +378,15 @@ mkldnn_memory_format_t GetDefaultFormat(const mkldnn::memory::desc &desc) {
case mkldnn_giohw:
case mkldnn_hwigo:
case mkldnn_hwigo_s8s8:
case mkldnn_gOIhw4i4o:
case mkldnn_gOIhw8i8o:
case mkldnn_gOIhw16i16o:
case mkldnn_gOIhw4i16o4i:
case mkldnn_gOIhw4i16o4i_s8s8:
case mkldnn_gOIhw8i16o2i:
case mkldnn_gOIhw8o16i2o:
case mkldnn_gOIhw8o8i:
case mkldnn_gOIhw4o4i:
case mkldnn_gOIhw16o16i:
case mkldnn_gIOhw16o16i:
case mkldnn_gOihw8o:
Expand Down