ArmNN
 20.11
RefReshapeWorkload.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "RefReshapeWorkload.hpp"
7 #include "RefWorkloadUtils.hpp"
8 #include "Profiling.hpp"
9 
10 #include <cstring>
11 
12 namespace armnn
13 {
14 
16 {
17  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefReshapeWorkload_Execute");
18 
19  void* output = GetOutputTensorData<void>(0, m_Data);
20  const void* input = GetInputTensorData<void>(0, m_Data);
21  unsigned int numBytes = GetTensorInfo(m_Data.m_Inputs[0]).GetNumBytes();
22  memcpy(output, input, numBytes);
23 }
24 
25 } //namespace armnn
CPU Execution: Reference C++ kernels.
const ReshapeQueueDescriptor m_Data
Definition: Workload.hpp:46
unsigned int GetNumBytes() const
Definition: Tensor.cpp:418
virtual void Execute() const override
Copyright (c) 2020 ARM Limited.
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:175
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers