First step in including hifimini kernels as part of the bazel build.
PiperOrigin-RevId: 305344175 Change-Id: I5d664f010faf40a9f76ddc07308723b683e7c2af
This commit is contained in:
parent
91cb053af0
commit
7b9f75751e
@ -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",
|
||||
"unpack.cc",
|
||||
] + select({
|
||||
"//conditions:default": [
|
||||
"conv.cc",
|
||||
"depthwise_conv.cc",
|
||||
"fully_connected.cc",
|
||||
"quantize.cc",
|
||||
"softmax.cc",
|
||||
"svdf.cc",
|
||||
"unpack.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)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user