Skip to content

Commit 1cfe3cc

Browse files
committed
修改SConscript结构
1 parent 7322f7b commit 1cfe3cc

3 files changed

Lines changed: 10 additions & 25 deletions

File tree

tensorflow/lite/micro/SConscript

Lines changed: 10 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,13 @@
11
from building import *
22
import os
33

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('''
511
kernels/activations.cc
612
kernels/arg_min_max.cc
713
kernels/ceil.cc
@@ -33,31 +39,10 @@ src = Split('''
3339
kernels/sub.cc
3440
kernels/svdf.cc
3541
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') \
3744
+ 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')
6146
src += Glob('tools/make/downloads/cmsis/CMSIS/NN/Source/ActivationFunctions/*.c')
6247
src += Glob('tools/make/downloads/cmsis/CMSIS/NN/Source/BasicMathFunctions/*.c')
6348
src += Glob('tools/make/downloads/cmsis/CMSIS/NN/Source/ConcatenationFunctions/*.c')

0 commit comments

Comments
 (0)