@@ -91,20 +91,43 @@ EXCLUDED_EXAMPLE_TESTS := \
91
91
MICRO_LITE_EXAMPLE_TESTS := $(filter -out $(EXCLUDED_EXAMPLE_TESTS ), $(MICRO_LITE_EXAMPLE_TESTS ))
92
92
MICRO_LITE_EXAMPLE_TESTS += $(shell find $(TENSORFLOW_ROOT )third_party /xtensa /examples / -name Makefile . inc )
93
93
94
- # Needed for LSTM support.
95
- MICROLITE_CC_KERNEL_SRCS := $(MICROLITE_CC_KERNEL_SRCS ) \
96
- $(TENSORFLOW_ROOT )tensorflow /lite /kernels /internal /reference /portable_tensor_utils . cc \
97
- $(TENSORFLOW_ROOT )tensorflow /lite /kernels /kernel_util . cc
98
-
99
94
ifeq ($(OPTIMIZED_KERNEL_DIR ), xtensa )
100
- MICROLITE_CC_KERNEL_SRCS := $(MICROLITE_CC_KERNEL_SRCS ) \
101
- $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /lstm_eval . cc \
102
- $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /lstm_eval_hifi . cc \
103
- $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /unidirectional_sequence_lstm . cc
95
+ # Explicitly add kernel sources specific to the Xtensa optimized
96
+ # implementations.
97
+ #
98
+ # Do not include overlays for reference kernel files.
99
+ MICROLITE_CC_KERNEL_SRCS += \
100
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /add_vision . cc \
101
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /conv_common_xtensa . cc \
102
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /conv_hifi . cc \
103
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /conv_int16_reference . cc \
104
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /conv_int8_int16 . cc \
105
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /conv_int8_reference . cc \
106
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /conv_vision . cc \
107
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /depthwise_conv_hifi . cc \
108
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /depthwise_conv_vision . cc \
109
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /fully_connected_common_xtensa . cc \
110
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /fully_connected_hifimini . cc \
111
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /fully_connected_int8 . cc \
112
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /fully_connected_vision . cc \
113
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /lstm_eval_hifi . cc \
114
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /pad_vision . cc \
115
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /pooling_int8 . cc \
116
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /pooling_vision . cc \
117
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /reduce_vision . cc \
118
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /reshape_vision . cc \
119
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /softmax_int8_int16 . cc \
120
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /softmax_vision . cc \
121
+ $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /svdf_hifimini . cc
122
+
123
+ # Needed for LSTM support.
124
+ MICROLITE_CC_KERNEL_SRCS += \
125
+ $(TENSORFLOW_ROOT )tensorflow /lite /kernels /internal /reference /portable_tensor_utils . cc \
126
+ $(TENSORFLOW_ROOT )tensorflow /lite /kernels /kernel_util . cc
127
+ endif
104
128
105
- # override KERNEL_OPTIMIZATION_LEVEL to enable higher performance
106
- # Xtensa intrinsics.
129
+ # override KERNEL_OPTIMIZATION_LEVEL to enable higher performance
130
+ # Xtensa intrinsics.
107
131
$(KERNEL_OBJDIR )$(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /decompress . o : $(TENSORFLOW_ROOT )tensorflow /lite /micro /kernels /xtensa /decompress . cc
108
132
@mkdir -p $(dir $@)
109
133
$(CXX ) $(CXXFLAGS ) -O3 -LNO :simd $(INCLUDES ) -c $< -o $@
110
- endif
0 commit comments