From 54557e413b411d4dcc50906f43fda7b269a25968 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Thu, 15 Feb 2024 09:09:40 +0100 Subject: [PATCH] Fix (core/shape): add view to enum (#850) --- src/brevitas/core/function_wrapper/shape.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/brevitas/core/function_wrapper/shape.py b/src/brevitas/core/function_wrapper/shape.py index 8b550f6f4..692ae76da 100644 --- 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