blob: ed5091299fb9621ec1bddb6a7a380c86dd499ef1 (
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* SPDX-FileCopyrightText: Copyright 2023 Arm Limited and/or its affiliates <open-source-office@arm.com>
* SPDX-License-Identifier: Apache-2.0
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "MicroMutableAllOpsResolver.hpp"
namespace arm {
namespace app {
/* Create our own AllOpsResolver by adding all Ops to MicroMutableOpResolver. */
tflite::MicroMutableOpResolver<kNumberOperators> CreateAllOpsResolver()
{
tflite::MicroMutableOpResolver<kNumberOperators> mutableAllOpResolver;
mutableAllOpResolver.AddAbs();
mutableAllOpResolver.AddAdd();
mutableAllOpResolver.AddAddN();
mutableAllOpResolver.AddArgMax();
mutableAllOpResolver.AddArgMin();
mutableAllOpResolver.AddAssignVariable();
mutableAllOpResolver.AddAveragePool2D();
mutableAllOpResolver.AddBatchToSpaceNd();
mutableAllOpResolver.AddBroadcastArgs();
mutableAllOpResolver.AddBroadcastTo();
mutableAllOpResolver.AddCallOnce();
mutableAllOpResolver.AddCast();
mutableAllOpResolver.AddCeil();
mutableAllOpResolver.AddCircularBuffer();
mutableAllOpResolver.AddConcatenation();
mutableAllOpResolver.AddConv2D();
mutableAllOpResolver.AddCos();
mutableAllOpResolver.AddCumSum();
mutableAllOpResolver.AddDepthToSpace();
mutableAllOpResolver.AddDepthwiseConv2D();
mutableAllOpResolver.AddDequantize();
mutableAllOpResolver.AddDetectionPostprocess();
mutableAllOpResolver.AddDiv();
mutableAllOpResolver.AddElu();
mutableAllOpResolver.AddEqual();
mutableAllOpResolver.AddEthosU();
mutableAllOpResolver.AddExp();
mutableAllOpResolver.AddExpandDims();
mutableAllOpResolver.AddFill();
mutableAllOpResolver.AddFloor();
mutableAllOpResolver.AddFloorDiv();
mutableAllOpResolver.AddFloorMod();
mutableAllOpResolver.AddFullyConnected();
mutableAllOpResolver.AddGather();
mutableAllOpResolver.AddGatherNd();
mutableAllOpResolver.AddGreater();
mutableAllOpResolver.AddGreaterEqual();
mutableAllOpResolver.AddHardSwish();
mutableAllOpResolver.AddIf();
mutableAllOpResolver.AddL2Normalization();
mutableAllOpResolver.AddL2Pool2D();
mutableAllOpResolver.AddLeakyRelu();
mutableAllOpResolver.AddLess();
mutableAllOpResolver.AddLessEqual();
mutableAllOpResolver.AddLog();
mutableAllOpResolver.AddLogicalAnd();
mutableAllOpResolver.AddLogicalNot();
mutableAllOpResolver.AddLogicalOr();
mutableAllOpResolver.AddLogistic();
mutableAllOpResolver.AddLogSoftmax();
mutableAllOpResolver.AddMaxPool2D();
mutableAllOpResolver.AddMaximum();
mutableAllOpResolver.AddMean();
mutableAllOpResolver.AddMinimum();
mutableAllOpResolver.AddMirrorPad();
mutableAllOpResolver.AddMul();
mutableAllOpResolver.AddNeg();
mutableAllOpResolver.AddNotEqual();
mutableAllOpResolver.AddPack();
mutableAllOpResolver.AddPad();
mutableAllOpResolver.AddPadV2();
mutableAllOpResolver.AddPrelu();
mutableAllOpResolver.AddQuantize();
mutableAllOpResolver.AddReadVariable();
mutableAllOpResolver.AddReduceMax();
mutableAllOpResolver.AddRelu();
mutableAllOpResolver.AddRelu6();
mutableAllOpResolver.AddReshape();
mutableAllOpResolver.AddResizeBilinear();
mutableAllOpResolver.AddResizeNearestNeighbor();
mutableAllOpResolver.AddRound();
mutableAllOpResolver.AddRsqrt();
mutableAllOpResolver.AddSelectV2();
mutableAllOpResolver.AddShape();
mutableAllOpResolver.AddSin();
mutableAllOpResolver.AddSlice();
mutableAllOpResolver.AddSoftmax();
mutableAllOpResolver.AddSpaceToBatchNd();
mutableAllOpResolver.AddSpaceToDepth();
mutableAllOpResolver.AddSplit();
mutableAllOpResolver.AddSplitV();
mutableAllOpResolver.AddSqrt();
mutableAllOpResolver.AddSquare();
mutableAllOpResolver.AddSquaredDifference();
mutableAllOpResolver.AddSqueeze();
mutableAllOpResolver.AddStridedSlice();
mutableAllOpResolver.AddSub();
mutableAllOpResolver.AddSum();
mutableAllOpResolver.AddSvdf();
mutableAllOpResolver.AddTanh();
mutableAllOpResolver.AddTranspose();
mutableAllOpResolver.AddTransposeConv();
mutableAllOpResolver.AddUnidirectionalSequenceLSTM();
mutableAllOpResolver.AddUnpack();
mutableAllOpResolver.AddVarHandle();
mutableAllOpResolver.AddWhile();
mutableAllOpResolver.AddZerosLike();
return mutableAllOpResolver;
}
} /* namespace app */
} /* namespace arm */
|
