EthosnRefPooling2dWorkload.cpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "EthosnRefPooling2dWorkload.hpp"

#include "EthosnRefPooling2d.hpp"
#include "EthosnRefWorkloadUtils.hpp"

using namespace armnn::ethosnref;

namespace armnn
{
void EthosnRefPooling2dWorkload::Execute() const
{
    ARMNN_SCOPED_PROFILING_EVENT_ETHOSN("EthosnRefPooling2dWorkload_Execute");

    const TensorInfo& inputInfo   = GetTensorInfo(m_Data.m_Inputs[0]);

    if (inputInfo.GetDataType() == DataType::QSymmS16) {
        const int16_t* inputData = GetInputTensorData<int16_t>(0, m_Data);

        EthosnRefPooling2d(m_Data, inputData);
    }
    else if (inputInfo.GetDataType() == DataType::QSymmS8) {
        const int8_t* inputData = GetInputTensorData<int8_t>(0, m_Data);

        EthosnRefPooling2d(m_Data, inputData);
    }
    else { // QAsymmU8
        assert(inputInfo.GetDataType() == DataType::QAsymmU8);

        const uint8_t* inputData = GetInputTensorData<uint8_t>(0, m_Data);

        EthosnRefPooling2d(m_Data, inputData);
    }
}
} //namespace armnn