ArmNN
 25.11
Loading...
Searching...
No Matches
RefDequantizeWorkload.cpp
Go to the documentation of this file.
1//
2// Copyright © 2019-2024 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
8#include "Encoders.hpp"
9#include "Decoders.hpp"
10#include "Dequantize.hpp"
11
12namespace armnn
13{
14
16{
17 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
18}
19
20void RefDequantizeWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const
21{
22 ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID("RefDequantizeWorkload_Execute");
23
24 const TensorInfo& inputInfo = GetTensorInfo(inputs[0]);
25 const TensorInfo& outputInfo = GetTensorInfo(outputs[0]);
26
27 auto inputDecoder = MakeDecoder<float>(inputInfo, inputs[0]->Map());
28 auto outputEncoder = MakeEncoder<float>(outputInfo, outputs[0]->Map());
29
30 Dequantize(*inputDecoder, *outputEncoder, inputInfo, outputInfo);
31}
32
33} // namespace armnn
#define ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID(label)
Creates a profiling event that uses GetGuid() and GetName() from the calling class.
Copyright (c) 2021 ARM Limited and Contributors.
std::unique_ptr< Decoder< T > > MakeDecoder(const TensorInfo &info, const void *data=nullptr)
std::unique_ptr< Encoder< T > > MakeEncoder(const TensorInfo &info, void *data=nullptr)
armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout, const armnn::DataType dataType)