diff options
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
| -rw-r--r-- | ethosu/vela/mark_tensors.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index b557f903..72ab8cfa 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -112,7 +112,9 @@ tensor_purposes = [ # ops, input_purpose ), ( set(("Conv2DBackpropInputSwitchedBias",)), - purpose_from_list([TensorPurpose.FeatureMap, TensorPurpose.Weights, TensorPurpose.FeatureMap, TensorPurpose.FeatureMap]), + purpose_from_list( + [TensorPurpose.FeatureMap, TensorPurpose.Weights, TensorPurpose.FeatureMap, TensorPurpose.FeatureMap] + ), ), ( set(("QuantizedConv2D", "QuantizedMatMul")), |