|
1 | 1 | from building import * |
2 | 2 | import os |
3 | 3 |
|
4 | | -src = Split(''' |
| 4 | +if GetDepend(['PKG_USING_TENSORFLOWLITEMICRO_REFERENCE']): |
| 5 | + src = Glob('kernels/*.cc') + Glob('*.c') + Glob('*.cc') + Glob('benchmarks/*.cc') \ |
| 6 | + + Glob('examples/micro_speech/*.cc') + Glob('examples/micro_speech/micro_features/*.cc') \ |
| 7 | + + Glob('memory_planner/*.cc') + Glob('testing/*.cc') + Glob('kernels/cmsis-nn/*.cc') |
| 8 | + |
| 9 | +if GetDepend(['PKG_USING_TENSORFLOWLITEMICRO_CMSISNN']): |
| 10 | + src = Split(''' |
5 | 11 | kernels/activations.cc |
6 | 12 | kernels/arg_min_max.cc |
7 | 13 | kernels/ceil.cc |
@@ -33,31 +39,10 @@ src = Split(''' |
33 | 39 | kernels/sub.cc |
34 | 40 | kernels/svdf.cc |
35 | 41 | kernels/tanh.cc |
36 | | - kernels/unpack.cc''') + Glob('*.c') + Glob('*.cc') + Glob('benchmarks/*.cc') \ |
| 42 | + kernels/unpack.cc |
| 43 | + ''') + Glob('*.c') + Glob('*.cc') + Glob('benchmarks/*.cc') \ |
37 | 44 | + Glob('examples/micro_speech/*.cc') + Glob('examples/micro_speech/micro_features/*.cc') \ |
38 | | - + Glob('memory_planner/*.cc') + Glob('testing/*.cc') |
39 | | - |
40 | | -if GetDepend(['PKG_USING_TENSORFLOWLITEMICRO_REFERENCE']): |
41 | | - src += Split(''' |
42 | | - kernels/add.cc |
43 | | - kernels/conv.cc |
44 | | - kernels/depthwise_conv.cc |
45 | | - kernels/fully_connected.cc |
46 | | - kernels/mul.cc |
47 | | - kernels/pooling.cc |
48 | | - kernels/softmax.cc |
49 | | - ''') |
50 | | - |
51 | | -if GetDepend(['PKG_USING_TENSORFLOWLITEMICRO_CMSISNN']): |
52 | | - src += Split(''' |
53 | | - kernels/cmsis-nn/add.cc |
54 | | - kernels/cmsis-nn/conv.cc |
55 | | - kernels/cmsis-nn/depthwise_conv.cc |
56 | | - kernels/cmsis-nn/fully_connected.cc |
57 | | - kernels/cmsis-nn/mul.cc |
58 | | - kernels/cmsis-nn/pooling.cc |
59 | | - kernels/cmsis-nn/softmax.cc |
60 | | - ''') |
| 45 | + + Glob('memory_planner/*.cc') + Glob('testing/*.cc') + Glob('kernels/cmsis-nn/*.cc') |
61 | 46 | src += Glob('tools/make/downloads/cmsis/CMSIS/NN/Source/ActivationFunctions/*.c') |
62 | 47 | src += Glob('tools/make/downloads/cmsis/CMSIS/NN/Source/BasicMathFunctions/*.c') |
63 | 48 | src += Glob('tools/make/downloads/cmsis/CMSIS/NN/Source/ConcatenationFunctions/*.c') |
|
0 commit comments