diff options
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index c80e18b5..af36587c 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -201,7 +201,7 @@ class Op(Enum): OneHot = OperatorInfo() Pack = OperatorInfo(indices=IFM_INDICES) PackReshaped = OperatorInfo(indices=IFM_INDICES) - Pad = OperatorInfo() + Pad = OperatorInfo(indices=IFM_INDICES) PadV2 = OperatorInfo() Placeholder = OperatorInfo() # Only used in CPU subgraphs Pow = OperatorInfo() @@ -335,6 +335,7 @@ class Op(Enum): class Padding(Enum): SAME = 0 VALID = 1 + EXPLICIT = 2 # Padding is specified in a PAD operation (only used for NPU operations) class ActivationFunction: |