Fix (export): shape propagation in debug flow

Xilinx · Nov 30, 2023 · fd7ce73 · fd7ce73
1 parent 3c48d06
commit fd7ce73
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