diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-05-20 09:03:40 +0200 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
| commit | cf7da10987cac3fc68cf180a9af665fe06d608fa (patch) | |
| tree | 5467fe8b062e4e37a4e0dadc178a9716d233facc /ethosu/vela/mark_tensors.py | |
| parent | 5b3dcd7323a5852bd188bb3929e7f38db3b52862 (diff) | |
| download | ethos-u-vela-cf7da10987cac3fc68cf180a9af665fe06d608fa.tar.gz | |
MLBEDSW-1716: Transpose Convolution support
Change-Id: Ie6d8d6de9f3447f19ba06aafa9fa480fc96a973b
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
| -rw-r--r-- | ethosu/vela/mark_tensors.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index bf7bc45f..5231e860 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -111,8 +111,8 @@ tensor_purposes = [ # ops, input_purpose purpose_from_list([TensorPurpose.FeatureMap, TensorPurpose.Weights, TensorPurpose.FeatureMap]), ), ( - set(("Conv2DBackpropInputSwitched",)), - purpose_from_list([TensorPurpose.FeatureMap, TensorPurpose.Weights, TensorPurpose.FeatureMap]), + set(("Conv2DBackpropInputSwitchedBias",)), + purpose_from_list([TensorPurpose.FeatureMap, TensorPurpose.Weights, TensorPurpose.FeatureMap, TensorPurpose.FeatureMap]), ), ( set(("QuantizedConv2D", "QuantizedMatMul")), |