Fixed Reshape

aidge_export_cpp/kernels/reshape.hpp

@@ -10,7 +10,7 @@ template<int M,
 __attribute__((always_inline)) inline
 void reshape_forward (
     const Input_T* __restrict inputs,
-    const Input_T* __restrict /*shape*/,
+    const int64_t* __restrict /*shape*/,
     Output_T* __restrict outputs)
 {
     // If inputs and outputs pointers are the same, the memory manager has already optimized this function so it is a no-op !

aidge_export_cpp/operators/Reshape.py

@@ -3,7 +3,7 @@ from aidge_core.export_utils import ExportNodeCpp
 from aidge_export_cpp import ROOT
 from aidge_export_cpp import ExportLibCpp
 
-@ExportLibCpp.register("Reshape", aidge_core.ImplSpec(aidge_core.IOSpec(aidge_core.dtype.float32)))
+@ExportLibCpp.register("Reshape", aidge_core.ImplSpec(aidge_core.IOSpec(aidge_core.dtype.any)))
 class ReshapeCPP(ExportNodeCpp):
     def __init__(self, node, mem_info):
         super().__init__(node, mem_info)