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/TosaMappings.hpp | |
| 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/TosaMappings.hpp')
| -rw-r--r-- | src/backends/tosaCommon/TosaMappings.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/tosaCommon/TosaMappings.hpp b/src/backends/tosaCommon/TosaMappings.hpp index c721bcaf59..7f137ca322 100644 --- a/src/backends/tosaCommon/TosaMappings.hpp +++ b/src/backends/tosaCommon/TosaMappings.hpp @@ -26,8 +26,9 @@ 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); // Function called in armnn::OptimizeSubgraphView() when access to armnn::Layer is available // and there is an option to set tosa basic block data from constant layer tenors available from the input layer. -TosaSerializationBasicBlock* GetTosaMappingFromLayer(Layer* layer); +TosaSerializationBasicBlock* GetTosaMappingFromLayer(Layer* layer, bool isMain); |