ArmNN
 25.11
Loading...
Searching...
No Matches
RefDepthToSpaceWorkload.cpp
Go to the documentation of this file.
1//
2// Copyright © 2019,2021-2024 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
7
8#include "DepthToSpace.hpp"
10
11namespace armnn
12{
13
15{
16 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
17}
18
19void RefDepthToSpaceWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const
20{
21 ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID("RefDepthToSpaceWorkload_Execute");
22 const TensorInfo inputInfo = GetTensorInfo(inputs[0]);
23
24 DepthToSpace(inputInfo,
26 inputs[0]->Map(),
27 outputs[0]->Map(),
28 GetDataTypeSize(inputInfo.GetDataType()));
29}
30
31} // namespace armnn
#define ARMNN_SCOPED_PROFILING_EVENT_REF_NAME_GUID(label)
Creates a profiling event that uses GetGuid() and GetName() from the calling class.
DataType GetDataType() const
Definition Tensor.hpp:200
Copyright (c) 2021 ARM Limited and Contributors.
constexpr unsigned int GetDataTypeSize(DataType dataType)
armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, unsigned int numberOfChannels, unsigned int height, unsigned int width, const armnn::DataLayout dataLayout, const armnn::DataType dataType)