diff options
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.cpp')
-rw-r--r-- | src/backends/tosaCommon/operatorMappings/TransposeConv2dOperator.cpp | 3 |
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); |