diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-07-27 13:23:15 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-07-28 13:57:09 +0000 |
commit | ec0959b02b3d4cfcb82deb12f5e952524bcb71ba (patch) | |
tree | b79aea0a558c00d7924493bde3e7b3a695d25dbd /shim/sl/canonical/Converter.cpp | |
parent | 40f6ea107f5dd829d69384ea71349771a804a1e1 (diff) | |
download | armnn-experimental/IVGCVSW-6964_Weight_Bias_ConstTensor.tar.gz |
IVGCVSW-7925 Add REVERSE V2 to Support Library (SL)experimental/IVGCVSW-6964_Weight_Bias_ConstTensor
* Fix typos in README
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6e30536d353fae7a7828d9e02e7301ab8dd8c115
Diffstat (limited to 'shim/sl/canonical/Converter.cpp')
-rw-r--r-- | shim/sl/canonical/Converter.cpp | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/shim/sl/canonical/Converter.cpp b/shim/sl/canonical/Converter.cpp index 790fad69dd..5b8c450733 100644 --- a/shim/sl/canonical/Converter.cpp +++ b/shim/sl/canonical/Converter.cpp @@ -150,6 +150,8 @@ bool Converter::ConvertOperation(const Operation& operation, const Model& model, return ConvertResize(operation, model, data, ResizeMethod::Bilinear); case OperationType::RESIZE_NEAREST_NEIGHBOR: return ConvertResize(operation, model, data, ResizeMethod::NearestNeighbor); + case OperationType::REVERSE: + return ConvertReverseV2(operation, model, data); case OperationType::RSQRT: return ConvertElementwiseUnary(operation, model, data, UnaryOperation::Rsqrt); case OperationType::SIN: @@ -4789,6 +4791,63 @@ bool Converter::ConvertResize(const Operation& operation, return SetupAndTrackLayerOutputSlot(operation, 0, *layer, model, data, nullptr, validateFunc); } +bool Converter::ConvertReverseV2(const Operation& operation, const Model& model, ConversionData& data) +{ + VLOG(DRIVER) << "Converter::ConvertReverseV2()"; + + LayerInputHandle input0 = ConvertToLayerInputHandle(operation, 0, model, data); + LayerInputHandle input1 = ConvertToLayerInputHandle(operation, 1, model, data); + if (!input0.IsValid() || !input1.IsValid()) + { + return Fail("%s: Operation has invalid inputs", __func__); + } + const armnn::TensorInfo& inputInfo0 = input0.GetTensorInfo(); + const armnn::TensorInfo& inputInfo1 = input1.GetTensorInfo(); + + const Operand* outputOperand = GetOutputOperand(operation, 0, model); + if (!outputOperand) + { + return Fail("%s: Could not read output 0", __func__); + } + const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*outputOperand); + + bool isSupported = false; + armnn::BackendId setBackend; + auto validateFunc = [&](const armnn::TensorInfo& outputInfo, bool& isSupported) + { + FORWARD_LAYER_SUPPORT_FUNC(__func__, + IsReverseV2Supported, + data.m_Backends, + isSupported, + setBackend, + inputInfo0, + inputInfo1, + outputInfo); + }; + + if(!IsDynamicTensor(outputInfo)) + { + validateFunc(outputInfo, isSupported); + } + else + { + isSupported = AreDynamicTensorsSupported(); + } + + if (!isSupported) + { + return false; + } + + armnn::IConnectableLayer* const layer = data.m_Network->AddReverseV2Layer(); + layer->SetBackendId(setBackend); + assert(layer != nullptr); + input0.Connect(layer->GetInputSlot(0)); + input1.Connect(layer->GetInputSlot(1)); + + return SetupAndTrackLayerOutputSlot(operation, 0, *layer, model, data, nullptr, validateFunc); +} + bool Converter::ConvertSpaceToBatchNd(const Operation& operation, const Model& model, ConversionData& data) { VLOG(DRIVER) << "Converter::ConvertSpaceToBatchNd()"; |