diff options
author | Charles Xu <charles.xu@arm.com> | 2020-05-04 11:32:05 +0200 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
commit | 53d4752c8dbd9152238abd854ec23e87e5123881 (patch) | |
tree | f52f6d781d8814bee73f18d7738ac6c49ae41022 /ethosu/vela/supported_operators.py | |
parent | 22df2ade766815fadb44addf46b5d78b81787b9d (diff) | |
download | ethos-u-vela-53d4752c8dbd9152238abd854ec23e87e5123881.tar.gz |
MLBEDSW-1649: Add size splits for Split op
The tensor is split into len(size_splits) along the dimension
axis with the sizes specified in the size_splits array.
Change-Id: I2ce98fa10e2e26f16cfd86a775aee94a308509ea
Signed-off-by: Charles Xu <charles.xu@arm.com>
Diffstat (limited to 'ethosu/vela/supported_operators.py')
-rw-r--r-- | ethosu/vela/supported_operators.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 70700e71..e5271450 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -73,7 +73,7 @@ class SupportedOperators: # bias add and batch norm | set(("QuantizedBiasAdd", "Requantize", "QuantizedBatchNorm", "BiasAdd", "FusedBatchNorm")) ) - self.split_ops = set(("Split", "StridedSlice", "Slice", "UnpackReshaped", "Unpack")) + self.split_ops = set(("Split", "SplitV", "StridedSlice", "Slice", "UnpackReshaped", "Unpack")) self.concat_ops = set(("Concat", "ConcatV2", "QuantizedConcat", "ConcatTFLite", "PackReshaped", "Pack")) self.memory_only_ops = ( set(("Squeeze", "Reshape", "QuantizedReshape", "ExpandDims")) | self.concat_ops | self.split_ops |