EthosnRefWorkloadUtils.hpp

Go to the documentation of this file.

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

#pragma once

#include <backendsCommon/CpuTensorHandle.hpp>

#include <armnn/Tensor.hpp>
#include <armnn/Types.hpp>

#include <ethosnref/EthosnRefTensorHandle.hpp>
#include <ethosnref/EthosnRefBackendId.hpp>

#include <Half.hpp>

#include <armnn/utility/PolymorphicDowncast.hpp>
#include <Profiling.hpp>

#define ARMNN_SCOPED_PROFILING_EVENT_ETHOSN(name) ARMNN_SCOPED_PROFILING_EVENT(armnn::EthosnRefBackendId(), name)

namespace armnn
{

namespace ethosnref
{

inline const TensorInfo& GetTensorInfo(const ITensorHandle* tensorHandle)
{
    // We know that ethosn workloads use EthosnRefTensorHandles for inputs and outputs
    const EthosnRefTensorHandle* refTensorHandle =
        PolymorphicDowncast<const EthosnRefTensorHandle*>(tensorHandle);
    return refTensorHandle->GetTensorInfo();
}

template <typename DataType, typename PayloadType>
const DataType* GetInputTensorData(unsigned int idx, const PayloadType& data)
{
    const ITensorHandle* tensorHandle = data.m_Inputs[idx];
    return reinterpret_cast<const DataType*>(tensorHandle->Map());
}

template <typename DataType, typename PayloadType>
DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data)
{
    ITensorHandle* tensorHandle = data.m_Outputs[idx];
    return reinterpret_cast<DataType*>(tensorHandle->Map());
}

inline uint32_t DivideRoundUp(uint32_t numerator, uint32_t denominator)
{
    return (numerator + denominator - 1) / denominator;
}

inline bool CheckDataType(DataType type, DataType inputType, DataType weightsType)
{
    return (inputType == type || weightsType == type);
}

}

} //namespace armnn