未验证 提交 c1802f91 编写于 作者: J jakpiase 提交者: GitHub

Fix for wrong results in segmentation models (#37310)

* fix

* ci rerun

* ci rerun

* ci Rerun
上级 521a274e
......@@ -230,6 +230,9 @@ inline dnnl::memory::format_tag GetMKLDNNFormat(dnnl::memory::desc mem_desc) {
} else if (strides[2] >= strides[3] && strides[3] >= strides[1] &&
strides[1] >= strides[0]) {
return dnnl::memory::format_tag::cdba;
} else if (strides[3] >= strides[2] && strides[2] >= strides[0] &&
strides[0] >= strides[1]) {
return dnnl::memory::format_tag::dcab;
} else {
return dnnl::memory::format_tag::nhwc;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册