blob: 6481a8a65305f243fa5a417b8344a172005bd711 [file] [log] [blame]
Cindy Liu43879e42023-10-18 11:18:03 -07001# Copyright 2023 Google LLC
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
Alex Van Dammed5554032023-08-01 15:39:38 -070015package(default_visibility = ["//visibility:public"])
16
17cc_library(
18 name = "opt",
19 srcs = [
Alex Van Dammef39cadd2023-08-28 16:21:20 -070020 "conv.cc",
Derek Chow16ffb602023-10-09 14:27:05 -070021 "depthwise_conv_s16.cc",
Alex Van Damme59112ad2023-08-11 11:10:33 -070022 "elementwise_add_s16.cc",
23 "elementwise_add_s32.cc",
Alex Van Damme7a2eaa42023-08-17 16:07:18 -070024 "elementwise_add_s8.cc",
25 "leaky_relu_s16.cc",
26 "leaky_relu_s8.cc",
Alex Van Dammed5554032023-08-01 15:39:38 -070027 "memcpy.cc",
Naveen-Dodda3285c6a2023-12-07 20:14:06 +000028 "max_pool_s8.cc"
Alex Van Dammed5554032023-08-01 15:39:38 -070029 ],
30 hdrs = [
31 "opt.h",
Alex Van Damme59112ad2023-08-11 11:10:33 -070032 "util.h",
Alex Van Dammed5554032023-08-01 15:39:38 -070033 ],
Alex Van Damme7a2eaa42023-08-17 16:07:18 -070034 target_compatible_with = ["@kelvin_sw//platforms/cpu:kelvin"],
Alex Van Dammed5554032023-08-01 15:39:38 -070035 deps = [
Cindy Liu43879e42023-10-18 11:18:03 -070036 "//crt",
Alex Van Dammef39cadd2023-08-28 16:21:20 -070037 "@tflite-micro//tensorflow/lite/kernels/internal:common",
Alex Van Dammed5554032023-08-01 15:39:38 -070038 ],
39 alwayslink = True,
Alex Van Dammed5554032023-08-01 15:39:38 -070040)