experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

First step in including hifimini kernels as part of the bazel build.

Browse Source 
PiperOrigin-RevId: 305344175
Change-Id: I5d664f010faf40a9f76ddc07308723b683e7c2af
This commit is contained in:
Mihai Maruseac 2020-04-07 14:43:13 -07:00 committed by TensorFlower Gardener
parent 91cb053af0
commit 7b9f75751e
1 changed files with 8 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
tensorflow/lite/micro/kernels/BUILD
View File

@ -14,11 +14,6 @@ package(
licenses = ["notice"], # Apache 2.0 licenses = ["notice"], # Apache 2.0
) )
config_setting(
name = "xtensa_hifimini",
define_values = {"tflm_build": "xtensa_hifimini"},
)
# LINT.IfChange(micro_ops) # LINT.IfChange(micro_ops)
cc_library( cc_library(
name = "micro_ops", name = "micro_ops",
@ -30,9 +25,12 @@ cc_library(
"circular_buffer.cc", "circular_buffer.cc",
"comparisons.cc", "comparisons.cc",
"concatenation.cc", "concatenation.cc",
"conv.cc",
"depthwise_conv.cc",
"dequantize.cc", "dequantize.cc",
"elementwise.cc", "elementwise.cc",
"floor.cc", "floor.cc",
"fully_connected.cc",
"logical.cc", "logical.cc",
"logistic.cc", "logistic.cc",
"maximum_minimum.cc", "maximum_minimum.cc",
@ -42,33 +40,17 @@ cc_library(
"pad.cc", "pad.cc",
"pooling.cc", "pooling.cc",
"prelu.cc", "prelu.cc",
"quantize.cc",
"reduce.cc", "reduce.cc",
"reshape.cc", "reshape.cc",
"round.cc", "round.cc",
"softmax.cc",
"split.cc", "split.cc",
"strided_slice.cc", "strided_slice.cc",
"sub.cc", "sub.cc",
"svdf.cc",
"unpack.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"], hdrs = ["micro_ops.h"],
copts = micro_copts(), copts = micro_copts(),
deps = [ deps = [
@ -85,10 +67,7 @@ cc_library(
"//tensorflow/lite/kernels/internal:tensor", "//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/kernels/internal:types", "//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/micro:micro_utils", "//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) # LINT.ThenChange(//tensorflow/lite/micro/kernels/BUILD:portable_optimized_micro_ops)