blob: 592f258b81711bebc5e23f5956b4baf9b178fac4 (
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "SerializerUtils.hpp"
namespace armnnSerializer
{
using namespace armnn;
armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType)
{
switch (dataType)
{
case armnn::DataType::Float32:
case armnn::DataType::Signed32:
return armnnSerializer::ConstTensorData::ConstTensorData_IntData;
case armnn::DataType::Float16:
return armnnSerializer::ConstTensorData::ConstTensorData_ShortData;
case armnn::DataType::QuantisedAsymm8:
case armnn::DataType::Boolean:
return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
default:
return armnnSerializer::ConstTensorData::ConstTensorData_NONE;
}
}
armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType)
{
switch (dataType)
{
case armnn::DataType::Float32:
return armnnSerializer::DataType::DataType_Float32;
case armnn::DataType::Float16:
return armnnSerializer::DataType::DataType_Float16;
case armnn::DataType::Signed32:
return armnnSerializer::DataType::DataType_Signed32;
case armnn::DataType::QuantisedAsymm8:
return armnnSerializer::DataType::DataType_QuantisedAsymm8;
case armnn::DataType::Boolean:
return armnnSerializer::DataType::DataType_Boolean;
default:
return armnnSerializer::DataType::DataType_Float16;
}
}
armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout)
{
switch (dataLayout)
{
case armnn::DataLayout::NHWC:
return armnnSerializer::DataLayout::DataLayout_NHWC;
case armnn::DataLayout::NCHW:
default:
return armnnSerializer::DataLayout::DataLayout_NCHW;
}
}
armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm)
{
switch (poolingAlgorithm)
{
case armnn::PoolingAlgorithm::Average:
return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
case armnn::PoolingAlgorithm::L2:
return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
case armnn::PoolingAlgorithm::Max:
default:
return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max;
}
}
armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding(armnn::OutputShapeRounding outputShapeRounding)
{
switch (outputShapeRounding)
{
case armnn::OutputShapeRounding::Ceiling:
return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Ceiling;
case armnn::OutputShapeRounding::Floor:
default:
return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Floor;
}
}
armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod paddingMethod)
{
switch (paddingMethod)
{
case armnn::PaddingMethod::IgnoreValue:
return armnnSerializer::PaddingMethod::PaddingMethod_IgnoreValue;
case armnn::PaddingMethod::Exclude:
default:
return armnnSerializer::PaddingMethod::PaddingMethod_Exclude;
}
}
} // namespace armnnSerializer
|
