aboutsummaryrefslogtreecommitdiff
path: root/src/backends/tosaCommon/TosaMappings.cpp
diff options
context:
space:
mode:
authorMatthew Sloyan <matthew.sloyan@arm.com>2022-11-09 16:28:51 +0000
committerTeresa Charlin <teresa.charlinreyes@arm.com>2022-11-09 19:27:49 +0000
commit5c54c3874ff5dc1656b9f28288e46086586f21b9 (patch)
tree81fefa759a5907faa92df55818fcc382ee67210a /src/backends/tosaCommon/TosaMappings.cpp
parent717a95ee6fb732bf22adee1e4f55b7ea30db5052 (diff)
downloadarmnn-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/TosaMappings.cpp')
-rw-r--r--src/backends/tosaCommon/TosaMappings.cpp10
1 files changed, 6 insertions, 4 deletions
diff --git a/src/backends/tosaCommon/TosaMappings.cpp b/src/backends/tosaCommon/TosaMappings.cpp
index 3c14bfd1f9..71d2012cbc 100644
--- a/src/backends/tosaCommon/TosaMappings.cpp
+++ b/src/backends/tosaCommon/TosaMappings.cpp
@@ -26,13 +26,14 @@ void SetBasicBlockConstantTensorData(Layer* layer, TosaSerializationBasicBlock*
TosaSerializationBasicBlock* GetTosaMapping(const LayerType type,
const std::vector<const TensorInfo*>& inputs,
const std::vector<const TensorInfo*>& outputs,
- const BaseDescriptor& /*descriptor*/)
+ const BaseDescriptor& /*descriptor*/,
+ bool isMain = false)
{
switch (type)
{
case LayerType::Addition:
{
- return ConvertAdditionToTosaOperator(inputs, outputs);
+ return ConvertAdditionToTosaOperator(inputs, outputs, isMain);
}
default:
{
@@ -43,7 +44,7 @@ TosaSerializationBasicBlock* GetTosaMapping(const LayerType type,
}
}
-TosaSerializationBasicBlock* GetTosaMappingFromLayer(Layer* layer)
+TosaSerializationBasicBlock* GetTosaMappingFromLayer(Layer* layer, bool isMain = false)
{
std::vector<const TensorInfo*> inputs;
for (auto inputSlot : layer->GetInputSlots())
@@ -60,7 +61,8 @@ TosaSerializationBasicBlock* GetTosaMappingFromLayer(Layer* layer)
TosaSerializationBasicBlock* basicBlock = GetTosaMapping(layer->GetType(),
inputs,
outputs,
- layer->GetParameters());
+ layer->GetParameters(),
+ isMain);
SetBasicBlockConstantTensorData(layer, basicBlock);
return basicBlock;
}