diff options
| author | Matthew Sloyan <matthew.sloyan@arm.com> | 2022-11-25 16:10:00 +0000 |
|---|---|---|
| committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2022-12-08 12:57:47 +0000 |
| commit | c5fe6e71cd39096af7c2523ec2afe96008c51b0c (patch) | |
| tree | 1486349bc36e17c1577465aab81d9eb3ca64e861 /src/backends/tosaCommon/operatorMappings/ConstantOperator.hpp | |
| parent | 3106c7f1714aea556d06d9f1e8c7faaeaeca996d (diff) | |
| download | armnn-c5fe6e71cd39096af7c2523ec2afe96008c51b0c.tar.gz | |
IVGCVSW-7168 Add Conv2d and Constant support to TOSA Reference Backend
* Added TOSA Conv2d and Constant mappings.
* Added unique naming to mappings based on previous and following
layers, so they are connected correctly.
* Updated existing mappings with new naming convention.
* Added all mappings to one main block in OptimizeSubgraphView.
* Removed isMain from mapping functions.
* Added Conv2d EndToEnd test.
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I27c3e238407c32379ce25a1f01dad11523ef5d2b
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/ConstantOperator.hpp')
| -rw-r--r-- | src/backends/tosaCommon/operatorMappings/ConstantOperator.hpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/ConstantOperator.hpp b/src/backends/tosaCommon/operatorMappings/ConstantOperator.hpp new file mode 100644 index 0000000000..df158aca3d --- /dev/null +++ b/src/backends/tosaCommon/operatorMappings/ConstantOperator.hpp @@ -0,0 +1,19 @@ +// +// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "TosaOperatorUtils.hpp" + +#include <Layer.hpp> + +#include <tosa_serialization_handler.h> + +using namespace armnn; +using namespace tosa; + +TosaSerializationBasicBlock* ConvertConstantToTosaOperator(const Layer* layer, + const std::vector<const TensorInfo*>& outputs); + |