diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-02-10 16:26:44 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-02-10 16:26:44 +0000 |
commit | 788e2c644e16fa7043b9a647806df46fd18bc040 (patch) | |
tree | 72b7f2a0f70f807c788dd7850f876ea1309872c2 /delegate/src/Round.hpp | |
parent | 29fde27f863ce82a2604270cfa7216904f1f171b (diff) | |
download | armnn-788e2c644e16fa7043b9a647806df46fd18bc040.tar.gz |
IVGCVSW-5400 'TfLiteDelegate: FLOOR operator support'
* Added FLOOR operator support to Arm NN TfLiteDelegate
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I986ce8c5a825f509e0f8b3d257fd5b60834c322f
Diffstat (limited to 'delegate/src/Round.hpp')
-rw-r--r-- | delegate/src/Round.hpp | 56 |
1 files changed, 48 insertions, 8 deletions
diff --git a/delegate/src/Round.hpp b/delegate/src/Round.hpp index 3335d0b337..1677607571 100644 --- a/delegate/src/Round.hpp +++ b/delegate/src/Round.hpp @@ -5,8 +5,6 @@ #pragma once -#include <armnn/utility/IgnoreUnused.hpp> - #include <tensorflow/lite/builtin_ops.h> #include <tensorflow/lite/c/builtin_op_data.h> #include <tensorflow/lite/c/common.h> @@ -21,13 +19,55 @@ TfLiteStatus VisitFloorOperator(DelegateData& delegateData, int nodeIndex, int32_t operatorCode) { - armnn::IgnoreUnused(delegateData, - tfLiteContext, - tfLiteNode, - nodeIndex, - operatorCode); + TF_LITE_ENSURE_STATUS(ValidateNumInputs(tfLiteContext, tfLiteNode, 1, nodeIndex)); + TF_LITE_ENSURE_STATUS(ValidateNumOutputs(tfLiteContext, tfLiteNode, 1, nodeIndex)); + + const TfLiteTensor* tfLiteTensors = tfLiteContext->tensors; + const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; + if (!IsValid(tfLiteContext, tfLiteInputTensor, operatorCode, nodeIndex)) + { + return kTfLiteError; + } + + const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]]; + if (!IsValid(tfLiteContext, tfLiteOutputTensor, operatorCode, nodeIndex)) + { + return kTfLiteError; + } + + const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); + const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor); + + bool isSupported = false; + auto validateFunc = [&](const armnn::TensorInfo& outInfo, bool& isSupported) + { + FORWARD_LAYER_SUPPORT_FUNC(__func__, + tfLiteContext, + IsFloorSupported, + delegateData.m_Backends, + isSupported, + inputTensorInfo, + outInfo); + }; + + // If the m_Network is a nullptr, this signals that a prerequisite TfLite callback is required to clarify the + // support for the operator + // If supported, VisitFloorOperator will be called again to add the layer to the network as seen further below + if (!delegateData.m_Network) + { + validateFunc(outputTensorInfo, isSupported); + return isSupported ? kTfLiteOk : kTfLiteError; + } + + // Add a Floor layer + armnn::IConnectableLayer* layer = delegateData.m_Network->AddFloorLayer(); + ARMNN_ASSERT(layer != nullptr); + + armnn::IOutputSlot& outputSlot = layer->GetOutputSlot(0); + outputSlot.SetTensorInfo(outputTensorInfo); - return kTfLiteError; + // Connect + return Connect(layer, tfLiteNode, delegateData); } } // namespace armnnDelegate |