Fix (export): shape propagation in debug flow

Xilinx · Nov 28, 2023 · f0a8c4f · f0a8c4f
1 parent 2d4ff9b
commit f0a8c4f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/brevitas/export/onnx/debug.py b/src/brevitas/export/onnx/debug.py
@@ -14,6 +14,7 @@ class DebugMarkerFunction(Function):
     @staticmethod
     def symbolic(g, input, export_debug_name):
         ret = g.op('brevitas.onnx::DebugMarker', input, export_debug_name_s=export_debug_name)
+        ret.setType(input.type())
         return ret
 
     @staticmethod