diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2022-11-09 16:28:51 +0000 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-11-09 19:27:49 +0000 |
commit | 5c54c3874ff5dc1656b9f28288e46086586f21b9 (patch) | |
tree | 81fefa759a5907faa92df55818fcc382ee67210a /src/backends/tosaCommon/operatorMappings | |
parent | 717a95ee6fb732bf22adee1e4f55b7ea30db5052 (diff) | |
download | armnn-5c54c3874ff5dc1656b9f28288e46086586f21b9.tar.gz |
IVGCVSW-7165 Implement TosaRefPreCompiledWorkload::Execute()
* Added FP32 support for TOSA Reference Backend.
* Added main block creation to OptimizeSubgraphView, this will only
occur once.
Change-Id: I169dac50b78e2c693da6327962c9f1d3ae3bd712
Signed-off-by: James Conroy <james.conroy@arm.com>
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings')
-rw-r--r-- | src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp | 9 | ||||
-rw-r--r-- | src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp | 3 |
2 files changed, 10 insertions, 2 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp b/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp index 98ea03ac3c..796797728e 100644 --- a/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp +++ b/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp @@ -6,7 +6,8 @@ #include "AdditionOperator.hpp" TosaSerializationBasicBlock* ConvertAdditionToTosaOperator(const std::vector<const TensorInfo*>& inputs, - const std::vector<const TensorInfo*>& outputs) + const std::vector<const TensorInfo*>& outputs, + bool isMain) { // A helper function with static global variables ensures uniqueness // for dynamically generating input, output and block names @@ -15,6 +16,12 @@ TosaSerializationBasicBlock* ConvertAdditionToTosaOperator(const std::vector<con std::string outputName = std::string("Op_ADD_output0_") + GetUniqueTosaMappingID(); std::string blockName = std::string("Op_ADD_block_") + GetUniqueTosaMappingID(); + // If it's the first block, overwrite block name with main. + if (isMain) + { + blockName = std::string("main"); + } + TosaSerializationOperator* op = new TosaSerializationOperator(Op_ADD, Attribute_NONE, nullptr, diff --git a/src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp b/src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp index 2a9c479d8e..f467bb6d10 100644 --- a/src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp +++ b/src/backends/tosaCommon/operatorMappings/AdditionOperator.hpp @@ -14,5 +14,6 @@ using namespace armnn; using namespace tosa; TosaSerializationBasicBlock* ConvertAdditionToTosaOperator(const std::vector<const TensorInfo*>& inputs, - const std::vector<const TensorInfo*>& outputs); + const std::vector<const TensorInfo*>& outputs, + bool isMain); |