add chanel first format support for rand=3 Deconv to use amx fp16 kernel

liubo-intel · Oct 23, 2024 · 85e45bc · 85e45bc
1 parent a4b7f35
commit 85e45bc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/plugins/intel_cpu/src/nodes/deconv.cpp b/src/plugins/intel_cpu/src/nodes/deconv.cpp
@@ -426,8 +426,12 @@ std::vector<memory::format_tag> Deconvolution::getAvailableFormatsForDims(const
     else if (dims.getRank() == 2)
         return {memory::format_tag::nc};
     else if (dims.getRank() == 3)
-        return {memory::format_tag::tnc, memory::format_tag::ntc,
-                memory::format_tag::ncw, memory::format_tag::nCw8c, memory::format_tag::nCw16c };
+        return {memory::format_tag::tnc,
+                memory::format_tag::ntc,
+                memory::format_tag::nwc,
+                memory::format_tag::ncw,
+                memory::format_tag::nCw8c,
+                memory::format_tag::nCw16c};
     else if (dims.getRank() == 4)
         return {memory::format_tag::nchw, memory::format_tag::nChw8c,
                 memory::format_tag::nChw16c, memory::format_tag::nhwc };