diff options
| author | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-11 17:19:54 +0100 |
|---|---|---|
| committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-22 15:57:43 +0100 |
| commit | ae2d553c4f3dd71a1df6c0e8c9cb920ae584b59e (patch) | |
| tree | 0b90f57e65cbc88b52b51b57cb9cc8230fa41ad2 /ethosu/vela/operation.py | |
| parent | 016b827ad722aecd4338d1d6c7b1b004760490b7 (diff) | |
| download | ethos-u-vela-ae2d553c4f3dd71a1df6c0e8c9cb920ae584b59e.tar.gz | |
MLBEDSW-3499: Support for PAD operator
Replaces the PAD operator by hardware padding when possible.
Change-Id: I9dce0885e51a4a73715824d7368637222e39b2b3
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
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: |