WIP adjust conv shape infer

maxnick · Oct 17, 2024 · 64c1584 · 64c1584
1 parent 23222d6
commit 64c1584
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/core/shape_inference/include/convolution_shape_inference_util.hpp b/src/core/shape_inference/include/convolution_shape_inference_util.hpp
@@ -274,8 +274,9 @@ void append_spatial_shape(const TOp* op,
                 pooling::valid_dilated_kernel_with_dim(op, filter_dilated.get_length(), dim, i);
             }
 
-            dim = dim::floor_div(dim - filter_dilated, strides[i]);
-            dim += 1;
+
+            dim = dim::floor_div(dim  + strides[i] - filter_dilated, strides[i]);
+            //dim += 1;
             out_shape.push_back(std::move(dim));
         }
     }