aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/NeonWorkloadUtils.hpp
blob: ec7688237ae59930ffeeaab13aaa49f8af8d2fb8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// See LICENSE file in the project root for full license information.
//
#pragma once

#include "Workload.hpp"

#include "backends/NeonTensorHandle.hpp"

#include "arm_compute/core/Types.h"
#include "arm_compute/core/Helpers.h"
#include "arm_compute/runtime/NEON/NEFunctions.h"
#include <arm_compute/runtime/SubTensor.h>

#include <boost/cast.hpp>

namespace armnn
{
class Layer;

template<typename T>
void InitialiseArmComputeTensorData(arm_compute::Tensor& tensor, const T* data);

} //namespace armnn