diff options
Diffstat (limited to 'delegate/src/Round.hpp')
-rw-r--r-- | delegate/src/Round.hpp | 71 |
1 files changed, 0 insertions, 71 deletions
diff --git a/delegate/src/Round.hpp b/delegate/src/Round.hpp deleted file mode 100644 index 7a060b1d8f..0000000000 --- a/delegate/src/Round.hpp +++ /dev/null @@ -1,71 +0,0 @@ -// -// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include "SharedFunctions.hpp" - -#include <tensorflow/lite/builtin_ops.h> -#include <tensorflow/lite/c/builtin_op_data.h> -#include <tensorflow/lite/c/common.h> -#include <tensorflow/lite/minimal_logging.h> - -namespace armnnDelegate -{ - -TfLiteStatus VisitFloorOperator(DelegateData& delegateData, - TfLiteContext* tfLiteContext, - TfLiteNode* tfLiteNode, - int nodeIndex, - int32_t 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); - // NOTE: looks like the outputTensorInfo is the only thing that is required for the case - // where we are adding the floor layer so maybe move the other stuff inside the - // if !delegateData block for efficiency. - const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true); - - // 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) - { - return ValidateFloorOperator(delegateData, tfLiteContext, inputTensorInfo, outputTensorInfo); - } - - // Add a Floor layer - armnn::IConnectableLayer* layer = delegateData.m_Network->AddFloorLayer(); - ARMNN_ASSERT(layer != nullptr); - - armnn::IOutputSlot& outputSlot = layer->GetOutputSlot(0); - outputSlot.SetTensorInfo(outputTensorInfo); - - // try to connect the Constant Inputs if there are any - if(ProcessInputs(layer,delegateData, tfLiteContext, tfLiteNode) != kTfLiteOk ) - { - return kTfLiteError; - } - - // Connect - return Connect(layer, tfLiteNode, delegateData); -} - -} // namespace armnnDelegate |