From 7b9f75751ebf889d1a76c7e01c4b4dfd7621cddc Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Tue, 7 Apr 2020 14:43:13 -0700 Subject: [PATCH] First step in including hifimini kernels as part of the bazel build. PiperOrigin-RevId: 305344175 Change-Id: I5d664f010faf40a9f76ddc07308723b683e7c2af --- tensorflow/lite/micro/kernels/BUILD | 37 +++++++---------------------- 1 file changed, 8 insertions(+), 29 deletions(-) diff --git a/tensorflow/lite/micro/kernels/BUILD b/tensorflow/lite/micro/kernels/BUILD index 9878c38c726..a0ffa342008 100644 --- a/tensorflow/lite/micro/kernels/BUILD +++ b/tensorflow/lite/micro/kernels/BUILD @@ -14,11 +14,6 @@ package( licenses = ["notice"], # Apache 2.0 ) -config_setting( - name = "xtensa_hifimini", - define_values = {"tflm_build": "xtensa_hifimini"}, -) - # LINT.IfChange(micro_ops) cc_library( name = "micro_ops", @@ -30,9 +25,12 @@ cc_library( "circular_buffer.cc", "comparisons.cc", "concatenation.cc", + "conv.cc", + "depthwise_conv.cc", "dequantize.cc", "elementwise.cc", "floor.cc", + "fully_connected.cc", "logical.cc", "logistic.cc", "maximum_minimum.cc", @@ -42,33 +40,17 @@ cc_library( "pad.cc", "pooling.cc", "prelu.cc", + "quantize.cc", "reduce.cc", "reshape.cc", "round.cc", + "softmax.cc", "split.cc", "strided_slice.cc", "sub.cc", + "svdf.cc", "unpack.cc", - ] + select({ - "//conditions:default": [ - "conv.cc", - "depthwise_conv.cc", - "fully_connected.cc", - "quantize.cc", - "softmax.cc", - "svdf.cc", - ], - "xtensa_hifimini": [ - "xtensa_hifimini/conv.cc", - "xtensa_hifimini/depthwise_conv.cc", - "xtensa_hifimini/fixedpoint_utils.h", - "xtensa_hifimini/fully_connected.cc", - "xtensa_hifimini/quantize.cc", - "xtensa_hifimini/softmax.cc", - "xtensa_hifimini/svdf.cc", - "xtensa_hifimini/utils.h", - ], - }), + ], hdrs = ["micro_ops.h"], copts = micro_copts(), deps = [ @@ -85,10 +67,7 @@ cc_library( "//tensorflow/lite/kernels/internal:tensor", "//tensorflow/lite/kernels/internal:types", "//tensorflow/lite/micro:micro_utils", - ] + select({ - "//conditions:default": [], - "xtensa_hifimini": ["//third_party/xtensa:hifi_mini_cstub64"], - }), + ], ) # LINT.ThenChange(//tensorflow/lite/micro/kernels/BUILD:portable_optimized_micro_ops)