[CPU] Fix: Add Dynamic logic of Col2Im (#33472)

src/plugins/intel_cpu/src/nodes/col2im.cpp

@@ -76,7 +76,40 @@ bool Col2Im::needPrepareParams() const {
 }
 
 void Col2Im::executeDynamicImpl(const dnnl::stream& strm) {
-    execute(strm);
+    // 1. get data shape
+    auto data_shape = getSrcMemoryAtPort(0)->getStaticDims();
+    size_t data_rank = data_shape.size();
+
+    // 2. get output_size
+    auto output_size_mem = getSrcMemoryAtPort(1);
+    const auto* output_size_ptr = output_size_mem->getDataAs<const int32_t>();
+
+    // 3. get kernel_size
+    auto kernel_size_mem = getSrcMemoryAtPort(2);
+    const auto* kernel_size_ptr = kernel_size_mem->getDataAs<const int32_t>();
+
+    // 4. calculate output_shape
+    size_t kernel_prod = static_cast<size_t>(kernel_size_ptr[0] * kernel_size_ptr[1]);
+
+    size_t H = static_cast<size_t>(output_size_ptr[0]);
+    size_t W = static_cast<size_t>(output_size_ptr[1]);
+
+    ov::Shape output_shape;
+    if (data_rank == 2) { // Case of Non-batched inputs
+        size_t C = data_shape[0] / kernel_prod;
+        output_shape = {C, H, W};
+        redefineOutputMemory({output_shape});
+        execute(strm);
+    }
+    else if (data_rank == 3) { // Case of Batched inputs
+        size_t N = data_shape[0];
+        size_t C = data_shape[1] / kernel_prod;
+        output_shape = {N, C, H, W};
+        redefineOutputMemory({output_shape});
+        execute(strm);
+    } else {
+        OPENVINO_THROW("Col2Im node supports only 2D(Non-Batched) or 3D(Batched) input tensors");
+    }
 }
 
 template <class T, class T_idx>

src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/classes/col2im.cpp

@@ -206,6 +206,78 @@ const std::vector<Col2ImSpecificParams> col2ImParamsVector = {
         ov::Strides{2, 2},
         ov::Shape{3, 3},
         ov::Shape{3, 3}
+    },
+    Col2ImSpecificParams { // Batched default
+        InputShape{{}, {{1, 4, 4}}},
+        std::vector<int64_t>{3, 3},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{1, 1},
+        ov::Strides{1, 1},
+        ov::Shape{0, 0},
+        ov::Shape{0, 0}
+    },
+    Col2ImSpecificParams { // Batched dilations
+        InputShape{{}, {{1, 4, 9}}},
+        std::vector<int64_t>{5, 5},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{1, 1},
+        ov::Strides{2, 2},
+        ov::Shape{0, 0},
+        ov::Shape{0, 0}
+    },
+    Col2ImSpecificParams { // Batched pads
+        InputShape{{}, {{1, 4, 9}}},
+        std::vector<int64_t>{2, 2},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{1, 1},
+        ov::Strides{1, 1},
+        ov::Shape{1, 1},
+        ov::Shape{1, 1}
+    },
+    Col2ImSpecificParams { // Batched strides
+        InputShape{{}, {{1, 4, 4}}},
+        std::vector<int64_t>{4, 4},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{2, 2},
+        ov::Strides{1, 1},
+        ov::Shape{0, 0},
+        ov::Shape{0, 0}
+    },
+    Col2ImSpecificParams { // Non-batched default
+        InputShape{{}, {{4, 4}}}, 
+        std::vector<int64_t>{3, 3},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{1, 1},
+        ov::Strides{1, 1},
+        ov::Shape{0, 0},
+        ov::Shape{0, 0}
+    },
+    Col2ImSpecificParams { // Non-batched dilations
+        InputShape{{}, {{4, 9}}},
+        std::vector<int64_t>{5, 5},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{1, 1},
+        ov::Strides{2, 2},
+        ov::Shape{0, 0},
+        ov::Shape{0, 0}
+    },
+    Col2ImSpecificParams { // Non-batched pads
+        InputShape{{}, {{4, 9}}},
+        std::vector<int64_t>{2, 2},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{1, 1},
+        ov::Strides{1, 1},
+        ov::Shape{1, 1},
+        ov::Shape{1, 1}
+    },
+    Col2ImSpecificParams { // Non-batched strides
+        InputShape{{}, {{4, 4}}},
+        std::vector<int64_t>{4, 4},
+        std::vector<int64_t>{2, 2},
+        ov::Strides{2, 2},
+        ov::Strides{1, 1},
+        ov::Shape{0, 0},
+        ov::Shape{0, 0}
     }
 };