-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathCMakeLists.txt
451 lines (436 loc) · 29.8 KB
/
CMakeLists.txt
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
#
# Copyright (C) 2021-2022 Chair of Electronic Design Automation, TUM.
#
# 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
#
# 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.
#
add_library(tflm STATIC)
#option(TF_LITE_STATIC_MEMORY "Turn off static memory" OFF)
# Enables the CMSIS-NN wrappers (make sure to use the correct TFLM source tree!)
option(USE_CMSIS_NN_WRAPPER "Build TensorFlow Lite Micro with CMSIS-NN wrapper" ON)
# Needed to run TensorFlow code with RISC-V toolchain
add_definitions(-DTF_LITE_USE_GLOBAL_MAX)
add_definitions(-DTF_LITE_USE_GLOBAL_MIN)
add_definitions(-DTF_LITE_STATIC_MEMORY)
add_definitions(-DTF_LITE_USE_GLOBAL_CMATH_FUNCTIONS)
add_definitions(-DTFLITE_EMULATE_FLOAT)
add_definitions(-fno-rtti) #Needed for GCC to resolve virtual delete function scope issue
add_definitions(-fno-exceptions) #Needed for GCC to resolve virtual delete function scope issue
#add_definitions(-DTF_LITE_STRIP_ERROR_STRINGS)
# TODO(fabianpedd): Figure out if these flags are actually needed
# -DTF_LITE_MCU_DEBUG_LOG
# -DTF_LITE_USE_CTIME
if(USE_CMSIS_NN_WRAPPER)
# Files for running with optimized CMSIS-NN wrapper (which muRISCV-NN uses)
add_definitions(-DCMSIS_NN)
set(TFLM_SRCS
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/fully_connected.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/mul.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/add.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/svdf.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/conv.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/depthwise_conv.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/pooling.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cmsis_nn/softmax.cc
)
else()
# Files for running with default TensorFlow kernels
set(TFLM_SRCS
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/fully_connected.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/mul.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/add.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/svdf.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/conv.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/depthwise_conv.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/pooling.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/softmax.cc
)
endif()
set(TFLM_SRCS
${TFLM_SRCS}
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/c/common.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/c/c_api_types.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/c/common.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/c/common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/c/builtin_op_data.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/builtin_ops.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/schema/schema_generated.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/compiler/mlir/lite/schema/schema_utils.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/schema/schema_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/context_util.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_resource_variable.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/fake_micro_context.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/test_helpers.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/micro_memory_planner.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/linear_memory_planner.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/memory_plan_struct.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/linear_memory_planner.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/non_persistent_buffer_planner_shim.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/greedy_memory_planner.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/non_persistent_buffer_planner_shim.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_planner/greedy_memory_planner.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_helpers.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/system_setup.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/memory_helpers.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/mock_micro_graph.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/system_setup.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/flatbuffer_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/fake_micro_context.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/recording_micro_allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_mutable_op_resolver.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/debug_log.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_profiler.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/test_helpers.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_resource_variable.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_context.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_profiler.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_utils.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_interpreter.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_allocation_info.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_string.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/mock_micro_graph.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/compatibility.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/flatbuffer_utils.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/tflite_bridge/micro_error_reporter.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_time.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_allocation_info.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_interpreter_graph.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_interpreter_context.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/single_arena_buffer_allocator.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/single_arena_buffer_allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/persistent_arena_buffer_allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/ibuffer_allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/persistent_arena_buffer_allocator.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/recording_single_arena_buffer_allocator.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/recording_single_arena_buffer_allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/non_persistent_arena_buffer_allocator.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/arena_allocator/non_persistent_arena_buffer_allocator.h
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_string.h
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/all_ops_resolver.h
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_graph.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/all_ops_resolver.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/recording_micro_interpreter.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_context.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_allocator.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/tflite_bridge/micro_error_reporter.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_interpreter.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_time.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/test_helper_custom_ops.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_graph.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/recording_micro_allocator.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/hard_swish.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/add_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/detection_postprocess_flexbuffers_generated_data.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/lstm_eval.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/dequantize.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/kernel_util.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/concatenation.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/space_to_batch_nd.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/resize_nearest_neighbor.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/softmax.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/pack.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/broadcast_args.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/maximum_minimum.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/mul_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/quantize.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/hard_swish_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/softmax_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cumsum.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/dequantize.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/prelu.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/pad.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/floor.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/split_v.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/svdf_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/logistic.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/shape.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/micro_tensor_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/conv_test.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/batch_to_space_nd.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/circular_buffer_flexbuffers_generated_data.h
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/micro_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/kernel_runner.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/leaky_relu.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/sub.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/logistic.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/unidirectional_sequence_lstm.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/floor_div.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/kernel_runner.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/hard_swish.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/ceil.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/strided_slice.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/depthwise_conv_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/logical.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/fully_connected.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/circular_buffer.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/pooling_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/micro_ops.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/kernel_util.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/assign_variable.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/sub_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/reduce.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/unpack.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/ethosu.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/activations_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/ethosu.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/conv.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/squared_difference.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/l2norm.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/l2_pool_2d.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/round.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/tanh.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/add_n.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/dequantize_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/broadcast_to.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/circular_buffer.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/space_to_depth.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/add.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/quantize_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/logical.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/split.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/if.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/expand_dims.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/logical_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/lstm_eval.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/sub.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/prelu_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/slice.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/gather.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/squeeze.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/floor_mod.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/depth_to_space.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/quantize.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/mul.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/fully_connected_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/gather_nd.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/reduce_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/conv_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/mirror_pad.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/activations.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/micro_tensor_utils.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/activations.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/fill.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/resize_bilinear.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/svdf.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/call_once.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/prelu.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/reshape.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/lstm_shared.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/zeros_like.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/neg.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/elementwise.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/read_variable.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/div.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/while.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/exp.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/reduce.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/detection_postprocess.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/depthwise_conv.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/circular_buffer_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/leaky_relu_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/activation_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/arg_min_max.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/comparisons.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/log_softmax.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/cast.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/leaky_relu.h
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/transpose_conv.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/pooling.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/elu.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/var_handle.cc
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/unidirectional_sequence_lstm_test_config.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/transpose.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/logistic_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/kernels/reshape_common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/debug_log.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_op_resolver.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_op_resolver.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_arena_constants.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/test_helper_custom_ops.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_log.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/micro_log.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/tflite_bridge/flatbuffer_conversions_bridge.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/micro/tflite_bridge/flatbuffer_conversions_bridge.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/compiler/mlir/lite/core/api/error_reporter.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/compiler/mlir/lite/core/api/error_reporter.h
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/api/op_resolver.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/api/flatbuffer_conversions.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/api/tensor_utils.h
# ${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/api/op_resolver.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/api/tensor_utils.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/core/api/flatbuffer_conversions.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/op_macros.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/kernel_util.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/kernel_util.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/tensor_ctypes.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/tensor_ctypes.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/strided_slice_logic.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/common.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/types.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/min.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/optimized/neon_check.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/portable_tensor.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/quantization_util.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/portable_tensor_utils_impl.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/softmax.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/slice.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/log_softmax.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/process_broadcast_shapes.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/broadcast_to.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/transpose.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/depthwiseconv_float.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/dequantize.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/prelu.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/logistic.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/comparisons.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/depthwiseconv_uint8.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/space_to_depth.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/ceil.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/sub.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/transpose_conv.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/depth_to_space.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/hard_swish.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/strided_slice.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/fully_connected.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/floor_mod.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/batch_to_space_nd.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/binary_function.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/requantize.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/conv.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/broadcast_args.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/concatenation.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/elu.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/add.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/pad.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/add_n.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/div.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/cumsum.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/exp.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/space_to_batch_nd.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/neg.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/batch_matmul.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/resize_nearest_neighbor.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/fill.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/quantize.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/mul.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/tanh.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/arg_min_max.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/portable_tensor_utils.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/l2normalization.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/round.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/reduce.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/portable_tensor_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/floor_div.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/leaky_relu.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/resize_bilinear.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/pooling.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/logistic.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/mean.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/transpose_conv.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/fully_connected.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/conv.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/add.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/mul.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/tanh.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/l2normalization.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/depthwise_conv.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/integer_ops/pooling.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/maximum_minimum.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/floor.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/reference/lstm_cell.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/compatibility.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/runtime_shape.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/max.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/quantization_util.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/portable_tensor_utils.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/portable_tensor_utils.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/cppmath.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/internal/common.cc
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/kernels/padding.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/portable_type_to_tflitetype.h
${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/lite/builtin_op_data.h
)
set(TFLM_THIRD_PARTY_SRCS
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/util.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/vector.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/flexbuffers.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/struct.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/array.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/buffer_ref.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/default_allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/verifier.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/buffer.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/vector_downward.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/flatbuffers.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/table.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/stl_emulation.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/base.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/flatbuffer_builder.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/string.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/allocator.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers/detached_buffer.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/gemmlowp/internal/detect_platform.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/gemmlowp/fixedpoint/fixedpoint_neon.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/gemmlowp/fixedpoint/fixedpoint_sse.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/gemmlowp/fixedpoint/fixedpoint.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/kissfft/tools/kiss_fftr.c
${CMAKE_CURRENT_SOURCE_DIR}/third_party/kissfft/tools/kiss_fftr.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/kissfft/_kiss_fft_guts.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/kissfft/kiss_fft.h
${CMAKE_CURRENT_SOURCE_DIR}/third_party/kissfft/kiss_fft.c
${CMAKE_CURRENT_SOURCE_DIR}/third_party/ruy/ruy/profiler/instrumentation.h
)
target_sources(tflm PRIVATE
${TFLM_SRCS}
${TFLM_THIRD_PARTY_SRCS}
)
target_include_directories(tflm PUBLIC
${CMAKE_CURRENT_SOURCE_DIR}
${CMAKE_CURRENT_SOURCE_DIR}/third_party/ruy
${CMAKE_CURRENT_SOURCE_DIR}/third_party/gemmlowp
${CMAKE_CURRENT_SOURCE_DIR}/third_party/kissfft
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers
${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include
${PROJECT_SOURCE_DIR}/Include
${PROJECT_SOURCE_DIR}/Include/CMSIS/NN
)
if(${TOOLCHAIN} STREQUAL "LLVM")
target_link_libraries(tflm PRIVATE
muriscvnn
m
stdc++
)
else()
target_link_libraries(tflm PRIVATE
muriscvnn
)
endif()
option(DISABLE_TFLM_AWW_INTG_TESTS "Enable tflm integration tests for aww model" OFF)
option(DISABLE_TFLM_IC_INTG_TESTS "Enable tflm integration tests for ic model" OFF)
option(DISABLE_TFLM_TOY_INTG_TESTS "Enable tflm integration tests for toy model" OFF)
option(DISABLE_TFLM_vWW_INTG_TESTS "Enable tflm integration tests for vww model" OFF)
if(NOT DISABLE_TFLM_AWW_INTG_TESTS)
add_subdirectory(aww)
endif()
if(NOT DISABLE_TFLM_IC_INTG_TESTS)
add_subdirectory(ic)
endif()
if(NOT DISABLE_TFLM_TOY_INTG_TESTS)
add_subdirectory(toy)
endif()
if(NOT DISABLE_TFLM_VWW_INTG_TESTS)
add_subdirectory(vww)
endif()