diff options
| author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-30 10:30:25 +0200 |
|---|---|---|
| committer | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-30 15:54:14 +0200 |
| commit | d85750702229af97c0b0bbda6e397a23254b6144 (patch) | |
| tree | 389962105a35d5cef595cfeb5d640bd59a0d0ff8 /ethosu/vela/pass_packing.py | |
| parent | cc5f4de1c35ba44fca7ff6295c6ae846f8242344 (diff) | |
| download | ethos-u-vela-d85750702229af97c0b0bbda6e397a23254b6144.tar.gz | |
Update version of Black to 22.3.0
Update version of Black to 22.3.0 due to updated dependencies.
Updates to fix reported issues due to new version.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: I60056aae452093ce8dcea1f499ecced22b25eef1
Diffstat (limited to 'ethosu/vela/pass_packing.py')
| -rw-r--r-- | ethosu/vela/pass_packing.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 1fefdf42..8535fa06 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -87,7 +87,14 @@ quantization_ops = set((Op.Dequantize, Op.Max, Op.Min)) cpu_ops = set((Op.Softmax, Op.LRN, Op.Shape, Op.Pad, Op.AddN)) | quantization_ops startup_init_ops = set((Op.Const, Op.Placeholder, Op.SubgraphInput)) -memory_only_ops = set((Op.Squeeze, Op.Reshape, Op.QuantizedReshape, Op.ExpandDims,)) +memory_only_ops = set( + ( + Op.Squeeze, + Op.Reshape, + Op.QuantizedReshape, + Op.ExpandDims, + ) +) test_sequence = [ |