diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2023-09-29 17:00:38 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2023-10-03 09:40:37 +0000 |
commit | ad323af0e9b47e53d366b85cdf74927f88748d40 (patch) | |
tree | c4251570a54ad9c123e4946bc1c69cbdbbf08253 /src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp | |
parent | edefbc282556021b7ea4db5d88d4cce2f9782b15 (diff) | |
download | armnn-ad323af0e9b47e53d366b85cdf74927f88748d40.tar.gz |
IVGCVSW-8033 Update Arm NN to use TOSA v0.60.0
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: If1504534fbd7d6b317fdb93083dcdbd8b827764b
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp')
-rw-r--r-- | src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp b/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp index a7af08347a..fdd6ca4c27 100644 --- a/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp +++ b/src/backends/tosaCommon/operatorMappings/Conv2dOperator.cpp @@ -114,7 +114,7 @@ TosaSerializationBasicBlock* ConvertConv2dToTosaOperator(const Layer* layer, static_cast<int>(conv2dDescriptor->m_StrideX)}; std::vector<int> dilation = {static_cast<int>(conv2dDescriptor->m_DilationY), static_cast<int>(conv2dDescriptor->m_DilationX)}; - TosaConvAttribute attribute(pad, stride, dilation, 0, 0, ArmNNToDType(inputs[0]->GetDataType())); + TosaConvAttribute attribute(pad, stride, dilation, 0, 0); auto* op = new TosaSerializationOperator(Op_CONV2D, Attribute_ConvAttribute, @@ -126,6 +126,7 @@ TosaSerializationBasicBlock* ConvertConv2dToTosaOperator(const Layer* layer, // operatorInputNames/operatorOutputNames ends up being the same as // blockInputNames/blockOutputNames for one-to-one ArmNN to TOSA mappings return new TosaSerializationBasicBlock(blockName, // name + mainName, // region name operators, // operators tensors, // tensors inputNames, // inputs |