Fix (core/shape): add view to enum (#850)

Xilinx · Feb 15, 2024 · 54557e4 · 54557e4
1 parent bdb121a
commit 54557e4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/brevitas/core/function_wrapper/shape.py b/src/brevitas/core/function_wrapper/shape.py
@@ -162,3 +162,4 @@ class StatsInputViewShapeImpl(object):
     OVER_OUTPUT_CHANNELS = OverOutputChannelView
     OVER_BATCH_OVER_TENSOR = OverBatchOverTensorView
     OVER_BATCH_OVER_OUTPUT_CHANNELS = OverBatchOverOutputChannelView
+    OVER_OUTPUT_FEATURES = OverOutputFeaturesView