ArmNN
NotReleased
ReshapeTestImpl.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include "
LayerTestResult.hpp
"
9
10
#include <
ResolveType.hpp
>
11
12
#include <
armnn/backends/IBackendInternal.hpp
>
13
#include <
backendsCommon/WorkloadFactory.hpp
>
14
15
template
<armnn::DataType ArmnnType,
typename
T = armnn::ResolveType<ArmnnType>>
16
LayerTestResult<T, 4>
SimpleReshapeTest
(
17
armnn::IWorkloadFactory
& workloadFactory,
18
const
armnn::IBackendInternal::IMemoryManagerSharedPtr
& memoryManager);
19
20
template
<armnn::DataType ArmnnType,
typename
T = armnn::ResolveType<ArmnnType>>
21
LayerTestResult<T, 5>
Reshape5dTest
(
22
armnn::IWorkloadFactory
& workloadFactory,
23
const
armnn::IBackendInternal::IMemoryManagerSharedPtr
& memoryManager);
WorkloadFactory.hpp
armnn::IWorkloadFactory
Definition:
WorkloadFactory.hpp:21
LayerTestResult
Definition:
LayerTestResult.hpp:29
LayerTestResult.hpp
armnn::IBackendInternal::IMemoryManagerSharedPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
Definition:
IBackendInternal.hpp:90
ResolveType.hpp
Reshape5dTest
LayerTestResult< T, 5 > Reshape5dTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition:
ReshapeTestImpl.cpp:109
IBackendInternal.hpp
SimpleReshapeTest
LayerTestResult< T, 4 > SimpleReshapeTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager)
Definition:
ReshapeTestImpl.cpp:57
src
backends
backendsCommon
test
layerTests
ReshapeTestImpl.hpp
Generated on Fri Mar 13 2020 16:06:56 for ArmNN by
1.8.13