11from building import *
22import os
33
4- if GetDepend (['PKG_USING_TENSORFLOWLITEMICRO_REFERENCE' ]):
5- src = Glob ('kernels/*.cc' ) + Glob ('*.c' ) + Glob ('*.cc' ) + Glob ('benchmarks/*.cc' ) \
6- + Glob ('examples/person_detection_experimental/*.cc' ) \
7- + Glob ('memory_planner/*.cc' ) + Glob ('testing/*.cc' ) + Glob ('kernels/reference/*.cc' ) \
8- + Glob ('tools/make/downloads/person_model_int8/*.cc' )
4+ src = Split ('''
5+ kernels/activations.cc
6+ kernels/arg_min_max.cc
7+ kernels/ceil.cc
8+ kernels/circular_buffer.cc
9+ kernels/comparisons.cc
10+ kernels/concatenation.cc
11+ kernels/dequantize.cc
12+ kernels/elementwise.cc
13+ kernels/ethosu.cc
14+ kernels/floor.cc
15+ kernels/hard_swish.cc
16+ kernels/kernel_runner.cc
17+ kernels/kernel_util.cc
18+ kernels/l2norm.cc
19+ kernels/logical.cc
20+ kernels/logistic.cc
21+ kernels/maximum_minimum.cc
22+ kernels/neg.cc
23+ kernels/pack.cc
24+ kernels/pad.cc
25+ kernels/prelu.cc
26+ kernels/quantize.cc
27+ kernels/reduce.cc
28+ kernels/reshape.cc
29+ kernels/resize_nearest_neighbor.cc
30+ kernels/round.cc
31+ kernels/split.cc
32+ kernels/strided_slice.cc
33+ kernels/sub.cc
34+ kernels/svdf.cc
35+ kernels/tanh.cc
36+ kernels/unpack.cc
37+ ''' ) + Glob ('*.c' ) + Glob ('*.cc' ) + Glob ('benchmarks/*.cc' ) \
38+ + Glob ('memory_planner/*.cc' ) + Glob ('testing/*.cc' )
939
10- if GetDepend (['PKG_USING_TENSORFLOWLITEMICRO_CMSISNN' ]):
11- src = Glob ('kernels/*.cc' ) + Glob ('*.c' ) + Glob ('*.cc' ) + Glob ('benchmarks/*.cc' ) \
12- + Glob ('examples/person_detection_experimental/*.cc' ) \
13- + Glob ('memory_planner/*.cc' ) + Glob ('testing/*.cc' ) + Glob ('kernels/cmsis-nn/*.cc' ) \
14- + Glob ('tools/make/downloads/person_model_int8/*.cc' )
40+ if GetDepend ('PKG_USING_TENSORFLOWLITEMICRO_AUDIO_EXAMPLE' ):
41+ src += Glob ('examples/micro_speech/*.cc' ) + Glob ('examples/micro_speech/micro_features/*.cc' )
1542
16- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/ActivationFunctions/*.c' )
17- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/BasicMathFunctions/*.c' )
18- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/ConcatenationFunctions/*.c' )
19- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/ConvolutionFunctions/*.c' )
20- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/FullyConnectedFunctions/*.c' )
21- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/NNSupportFunctions/*.c' )
22- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/PoolingFunctions/*.c' )
23- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/ReshapeFunctions/*.c' )
24- src += Glob ('tools/make/downloads/cmsis/CMSIS/NN/Source/SoftmaxFunctions/*.c' )
43+ if GetDepend ('PKG_USING_TENSORFLOWLITEMICRO_REFERENCE' ):
44+ src += Split ('''
45+ kernels/add.cc
46+ kernels/conv.cc
47+ kernels/depthwise_conv.cc
48+ kernels/fully_connected.cc
49+ kernels/mul.cc
50+ kernels/pooling.cc
51+ kernels/softmax.cc
52+ ''' )
53+ if GetDepend ('PKG_USING_TENSORFLOWLITEMICRO_CMSISNN' ):
54+ src += Split ('''
55+ kernels/cmsis-nn/add.cc
56+ kernels/cmsis-nn/conv.cc
57+ kernels/cmsis-nn/depthwise_conv.cc
58+ kernels/cmsis-nn/fully_connected.cc
59+ kernels/cmsis-nn/mul.cc
60+ kernels/cmsis-nn/pooling.cc
61+ kernels/cmsis-nn/softmax.cc
62+ ''' )
63+ src += Glob ('tools/cmsis/CMSIS/NN/Source/ActivationFunctions/*.c' )
64+ src += Glob ('tools/cmsis/CMSIS/NN/Source/BasicMathFunctions/*.c' )
65+ src += Glob ('tools/cmsis/CMSIS/NN/Source/ConcatenationFunctions/*.c' )
66+ src += Glob ('tools/cmsis/CMSIS/NN/Source/ConvolutionFunctions/*.c' )
67+ src += Glob ('tools/cmsis/CMSIS/NN/Source/FullyConnectedFunctions/*.c' )
68+ src += Glob ('tools/cmsis/CMSIS/NN/Source/NNSupportFunctions/*.c' )
69+ src += Glob ('tools/cmsis/CMSIS/NN/Source/PoolingFunctions/*.c' )
70+ src += Glob ('tools/cmsis/CMSIS/NN/Source/ReshapeFunctions/*.c' )
71+ src += Glob ('tools/cmsis/CMSIS/NN/Source/SoftmaxFunctions/*.c' )
2572
2673#.
2774root = str (Dir ('#' ))
@@ -40,7 +87,7 @@ kissfft = os.path.join(tflm_pkg, "third_party/kissfft")
4087#./third_party/ruy
4188ruy = os .path .join (tflm_pkg , "third_party/ruy" )
4289#./tensorflow/lite/micro/tools/make/downloads
43- cmsis = os .path .join (tflm_pkg , "tensorflow/lite/micro/tools/make/downloads " )
90+ cmsis = os .path .join (tflm_pkg , "tensorflow/lite/micro/tools/" )
4491
4592
4693CPPPATH = [tflm_pkg , flatbuffer , gemmlowp , kissfft , ruy , cmsis ]
0 commit comments