blob: a9e945608c382d3c1e8ef8717528580fbb8a85ec [file] [log] [blame]
Lev Proleevb38bb4f2020-12-15 19:25:32 +00001/*
2 * Copyright (C) 2021 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <aidl/android/hardware/neuralnetworks/IPreparedModel.h>
18#include <aidl/android/hardware/neuralnetworks/OperandType.h>
19#include <aidl/android/hardware/neuralnetworks/OperationType.h>
20
21#include <ControlFlow.h>
22#include <TestHarness.h>
23
24namespace aidl::android::hardware::neuralnetworks {
25
26namespace nn = ::android::nn;
27
28static_assert(static_cast<uint64_t>(IPreparedModel::DEFAULT_LOOP_TIMEOUT_DURATION_NS) ==
29 nn::operation_while::kTimeoutNsDefault);
30static_assert(static_cast<uint64_t>(IPreparedModel::MAXIMUM_LOOP_TIMEOUT_DURATION_NS) ==
31 nn::operation_while::kTimeoutNsMaximum);
32
33// Make sure that the HIDL enums are compatible with the values defined in
34// frameworks/ml/nn/tools/test_generator/test_harness/include/TestHarness.h.
35using namespace test_helper;
36#define CHECK_TEST_ENUM(EnumType, enumValue) \
37 static_assert(static_cast<EnumType>(Test##EnumType::enumValue) == EnumType::enumValue)
38
39CHECK_TEST_ENUM(OperandType, FLOAT32);
40CHECK_TEST_ENUM(OperandType, INT32);
41CHECK_TEST_ENUM(OperandType, UINT32);
42CHECK_TEST_ENUM(OperandType, TENSOR_FLOAT32);
43CHECK_TEST_ENUM(OperandType, TENSOR_INT32);
44CHECK_TEST_ENUM(OperandType, TENSOR_QUANT8_ASYMM);
45CHECK_TEST_ENUM(OperandType, BOOL);
46CHECK_TEST_ENUM(OperandType, TENSOR_QUANT16_SYMM);
47CHECK_TEST_ENUM(OperandType, TENSOR_FLOAT16);
48CHECK_TEST_ENUM(OperandType, TENSOR_BOOL8);
49CHECK_TEST_ENUM(OperandType, FLOAT16);
50CHECK_TEST_ENUM(OperandType, TENSOR_QUANT8_SYMM_PER_CHANNEL);
51CHECK_TEST_ENUM(OperandType, TENSOR_QUANT16_ASYMM);
52CHECK_TEST_ENUM(OperandType, TENSOR_QUANT8_SYMM);
53CHECK_TEST_ENUM(OperandType, TENSOR_QUANT8_ASYMM_SIGNED);
54
55CHECK_TEST_ENUM(OperationType, ADD);
56CHECK_TEST_ENUM(OperationType, AVERAGE_POOL_2D);
57CHECK_TEST_ENUM(OperationType, CONCATENATION);
58CHECK_TEST_ENUM(OperationType, CONV_2D);
59CHECK_TEST_ENUM(OperationType, DEPTHWISE_CONV_2D);
60CHECK_TEST_ENUM(OperationType, DEPTH_TO_SPACE);
61CHECK_TEST_ENUM(OperationType, DEQUANTIZE);
62CHECK_TEST_ENUM(OperationType, EMBEDDING_LOOKUP);
63CHECK_TEST_ENUM(OperationType, FLOOR);
64CHECK_TEST_ENUM(OperationType, FULLY_CONNECTED);
65CHECK_TEST_ENUM(OperationType, HASHTABLE_LOOKUP);
66CHECK_TEST_ENUM(OperationType, L2_NORMALIZATION);
67CHECK_TEST_ENUM(OperationType, L2_POOL_2D);
68CHECK_TEST_ENUM(OperationType, LOCAL_RESPONSE_NORMALIZATION);
69CHECK_TEST_ENUM(OperationType, LOGISTIC);
70CHECK_TEST_ENUM(OperationType, LSH_PROJECTION);
71CHECK_TEST_ENUM(OperationType, LSTM);
72CHECK_TEST_ENUM(OperationType, MAX_POOL_2D);
73CHECK_TEST_ENUM(OperationType, MUL);
74CHECK_TEST_ENUM(OperationType, RELU);
75CHECK_TEST_ENUM(OperationType, RELU1);
76CHECK_TEST_ENUM(OperationType, RELU6);
77CHECK_TEST_ENUM(OperationType, RESHAPE);
78CHECK_TEST_ENUM(OperationType, RESIZE_BILINEAR);
79CHECK_TEST_ENUM(OperationType, RNN);
80CHECK_TEST_ENUM(OperationType, SOFTMAX);
81CHECK_TEST_ENUM(OperationType, SPACE_TO_DEPTH);
82CHECK_TEST_ENUM(OperationType, SVDF);
83CHECK_TEST_ENUM(OperationType, TANH);
84CHECK_TEST_ENUM(OperationType, BATCH_TO_SPACE_ND);
85CHECK_TEST_ENUM(OperationType, DIV);
86CHECK_TEST_ENUM(OperationType, MEAN);
87CHECK_TEST_ENUM(OperationType, PAD);
88CHECK_TEST_ENUM(OperationType, SPACE_TO_BATCH_ND);
89CHECK_TEST_ENUM(OperationType, SQUEEZE);
90CHECK_TEST_ENUM(OperationType, STRIDED_SLICE);
91CHECK_TEST_ENUM(OperationType, SUB);
92CHECK_TEST_ENUM(OperationType, TRANSPOSE);
93CHECK_TEST_ENUM(OperationType, ABS);
94CHECK_TEST_ENUM(OperationType, ARGMAX);
95CHECK_TEST_ENUM(OperationType, ARGMIN);
96CHECK_TEST_ENUM(OperationType, AXIS_ALIGNED_BBOX_TRANSFORM);
97CHECK_TEST_ENUM(OperationType, BIDIRECTIONAL_SEQUENCE_LSTM);
98CHECK_TEST_ENUM(OperationType, BIDIRECTIONAL_SEQUENCE_RNN);
99CHECK_TEST_ENUM(OperationType, BOX_WITH_NMS_LIMIT);
100CHECK_TEST_ENUM(OperationType, CAST);
101CHECK_TEST_ENUM(OperationType, CHANNEL_SHUFFLE);
102CHECK_TEST_ENUM(OperationType, DETECTION_POSTPROCESSING);
103CHECK_TEST_ENUM(OperationType, EQUAL);
104CHECK_TEST_ENUM(OperationType, EXP);
105CHECK_TEST_ENUM(OperationType, EXPAND_DIMS);
106CHECK_TEST_ENUM(OperationType, GATHER);
107CHECK_TEST_ENUM(OperationType, GENERATE_PROPOSALS);
108CHECK_TEST_ENUM(OperationType, GREATER);
109CHECK_TEST_ENUM(OperationType, GREATER_EQUAL);
110CHECK_TEST_ENUM(OperationType, GROUPED_CONV_2D);
111CHECK_TEST_ENUM(OperationType, HEATMAP_MAX_KEYPOINT);
112CHECK_TEST_ENUM(OperationType, INSTANCE_NORMALIZATION);
113CHECK_TEST_ENUM(OperationType, LESS);
114CHECK_TEST_ENUM(OperationType, LESS_EQUAL);
115CHECK_TEST_ENUM(OperationType, LOG);
116CHECK_TEST_ENUM(OperationType, LOGICAL_AND);
117CHECK_TEST_ENUM(OperationType, LOGICAL_NOT);
118CHECK_TEST_ENUM(OperationType, LOGICAL_OR);
119CHECK_TEST_ENUM(OperationType, LOG_SOFTMAX);
120CHECK_TEST_ENUM(OperationType, MAXIMUM);
121CHECK_TEST_ENUM(OperationType, MINIMUM);
122CHECK_TEST_ENUM(OperationType, NEG);
123CHECK_TEST_ENUM(OperationType, NOT_EQUAL);
124CHECK_TEST_ENUM(OperationType, PAD_V2);
125CHECK_TEST_ENUM(OperationType, POW);
126CHECK_TEST_ENUM(OperationType, PRELU);
127CHECK_TEST_ENUM(OperationType, QUANTIZE);
128CHECK_TEST_ENUM(OperationType, QUANTIZED_16BIT_LSTM);
129CHECK_TEST_ENUM(OperationType, RANDOM_MULTINOMIAL);
130CHECK_TEST_ENUM(OperationType, REDUCE_ALL);
131CHECK_TEST_ENUM(OperationType, REDUCE_ANY);
132CHECK_TEST_ENUM(OperationType, REDUCE_MAX);
133CHECK_TEST_ENUM(OperationType, REDUCE_MIN);
134CHECK_TEST_ENUM(OperationType, REDUCE_PROD);
135CHECK_TEST_ENUM(OperationType, REDUCE_SUM);
136CHECK_TEST_ENUM(OperationType, ROI_ALIGN);
137CHECK_TEST_ENUM(OperationType, ROI_POOLING);
138CHECK_TEST_ENUM(OperationType, RSQRT);
139CHECK_TEST_ENUM(OperationType, SELECT);
140CHECK_TEST_ENUM(OperationType, SIN);
141CHECK_TEST_ENUM(OperationType, SLICE);
142CHECK_TEST_ENUM(OperationType, SPLIT);
143CHECK_TEST_ENUM(OperationType, SQRT);
144CHECK_TEST_ENUM(OperationType, TILE);
145CHECK_TEST_ENUM(OperationType, TOPK_V2);
146CHECK_TEST_ENUM(OperationType, TRANSPOSE_CONV_2D);
147CHECK_TEST_ENUM(OperationType, UNIDIRECTIONAL_SEQUENCE_LSTM);
148CHECK_TEST_ENUM(OperationType, UNIDIRECTIONAL_SEQUENCE_RNN);
149CHECK_TEST_ENUM(OperationType, RESIZE_NEAREST_NEIGHBOR);
150
151#undef CHECK_TEST_ENUM
152
153} // namespace aidl::android::hardware::neuralnetworks