Fix intel#28 error (Update pooling primitive)

zhuhaozhe · Apr 8, 2022 · 3797cf0 · 3797cf0
1 parent 8d94406
commit 3797cf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/ideep/operators/pool.hpp b/include/ideep/operators/pool.hpp
@@ -108,7 +108,7 @@ struct pooling_backward : public dnnl::pooling_backward {
                       const dims& padding_r,
                       algorithm aalgorithm,
                       const engine& aengine = engine::cpu_engine()) {
-    auto src_desc = src.get_desc().to_format_any();
+    auto src_desc = src.get_desc();
     auto dst_desc = dst.get_desc();
 
     auto forward_hints =