blob: 3464e0de9b2bb27fdc95702b36fd0fa70c394a35 [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 = [
Lun Dong3b8d3cb2024-05-07 01:50:35 -070020 "conv_s16_b32.cc",
21 "conv_s16_b64.cc",
Alex Van Dammeb11ad172024-04-18 14:18:17 -070022 "conv_s8.cc",
Lun Dong3b8d3cb2024-05-07 01:50:35 -070023 "conv_s8_1x1.cc",
24 "conv_s8_3x1_d48.cc",
Alex Van Dammee617f612024-05-08 16:41:37 -070025 "conv_s8_d4.cc",
Lun Dong3b8d3cb2024-05-07 01:50:35 -070026 "conv_s8_d32.cc",
Derek Chow86806b32024-05-09 15:26:01 -070027 "conv_s8_d1.cc",
Derek Chow16ffb602023-10-09 14:27:05 -070028 "depthwise_conv_s16.cc",
Naveen Doddabe4ab972024-04-17 17:47:46 +000029 "depthwise_conv_s8.cc",
Alex Van Damme59112ad2023-08-11 11:10:33 -070030 "elementwise_add_s16.cc",
31 "elementwise_add_s32.cc",
Alex Van Damme7a2eaa42023-08-17 16:07:18 -070032 "elementwise_add_s8.cc",
33 "leaky_relu_s16.cc",
34 "leaky_relu_s8.cc",
Lun Dong3b8d3cb2024-05-07 01:50:35 -070035 "max_pool_s8.cc",
Alex Van Dammed5554032023-08-01 15:39:38 -070036 "memcpy.cc",
37 ],
38 hdrs = [
Lun Dong3b8d3cb2024-05-07 01:50:35 -070039 "conv_s8.h",
40 "conv_util.h",
Alex Van Dammed5554032023-08-01 15:39:38 -070041 "opt.h",
Alex Van Damme59112ad2023-08-11 11:10:33 -070042 "util.h",
Alex Van Dammed5554032023-08-01 15:39:38 -070043 ],
Alex Van Damme7a2eaa42023-08-17 16:07:18 -070044 target_compatible_with = ["@kelvin_sw//platforms/cpu:kelvin"],
Alex Van Dammed5554032023-08-01 15:39:38 -070045 deps = [
Cindy Liu43879e42023-10-18 11:18:03 -070046 "//crt",
Alex Van Dammef39cadd2023-08-28 16:21:20 -070047 "@tflite-micro//tensorflow/lite/kernels/internal:common",
Naveen Doddabe4ab972024-04-17 17:47:46 +000048 "@tflite-micro//tensorflow/lite/kernels/internal:reference_base",
Alex Van Dammed5554032023-08-01 15:39:38 -070049 ],
50 alwayslink = True,
Alex Van Dammed5554032023-08-01 15:39:38 -070051)