diff options
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 252f03b7..14818870 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -152,7 +152,7 @@ input and output tensors, as well as an attribute dictionary.""" weight_idx = 1 ofm_idx = 0 - elif self.type in ("Squeeze", "Reshape", "QuantizedReshape", "ExpandDims"): + elif self.type in ("Squeeze", "Reshape", "QuantizedReshape", "ExpandDims", "Sigmoid", "Tanh"): ifm_idx = 0 ofm_idx = 0 |