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

Add MKLDNN 4c layout to fix gluoncv se_resnext101_64x4d #15692

Merged
merged 2 commits into from
Aug 1, 2019
Merged

Add MKLDNN 4c layout to fix gluoncv se_resnext101_64x4d #15692

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
399dafb
add 4c type
xinyu-intel Jul 29, 2019
d8ff28e
trigger
xinyu-intel Jul 31, 2019
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