ArmNN
 26.01
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
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.
QueueDescriptor m_Data
Definition Workload.hpp:74
Copyright (c) 2021 ARM Limited and Contributors.
armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout, const armnn::DataType dataType)
std::vector< ITensorHandle * > m_Inputs
std::vector< ITensorHandle * > m_Outputs