From cf7da10987cac3fc68cf180a9af665fe06d608fa Mon Sep 17 00:00:00 2001 From: Jacob Bohlin Date: Wed, 20 May 2020 09:03:40 +0200 Subject: MLBEDSW-1716: Transpose Convolution support Change-Id: Ie6d8d6de9f3447f19ba06aafa9fa480fc96a973b Signed-off-by: Jacob Bohlin --- ethosu/vela/mark_tensors.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/mark_tensors.py') 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")), -- cgit v1.2.1