1 files changed, 1 insertions, 2 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.cpp b/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.cpp
index a0d58e2fa8..1ad8c9562f 100644
--- a/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.cpp
+++ b/src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.cpp
@@ -94,8 +94,7 @@ TosaSerializationBasicBlock* ConvertTransposeConv2dToTosaOperator(const Layer* l
     {
         // If bias is disabled, create a constant bias tensor of 0's as three inputs are required.
         // The size of the bias must match the channels dimension, so get the correct index.
-        unsigned int index = (descriptor->m_DataLayout == DataLayout::NHWC) ?
-                outputs[0]->GetShape()[3] : outputs[0]->GetShape()[1];
+        unsigned int index = (descriptor->m_DataLayout == DataLayout::NHWC) ? 3 : 1;
 
         std::vector<uint8_t> uint8Data;
         std::vector<float> data(outputs[0]->GetShape()[index], 0.0f);