Cindy Liu | 43879e4 | 2023-10-18 11:18:03 -0700 | [diff] [blame] | 1 | # 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 Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 15 | package(default_visibility = ["//visibility:public"]) |
| 16 | |
| 17 | cc_library( |
| 18 | name = "opt", |
| 19 | srcs = [ |
Lun Dong | 3b8d3cb | 2024-05-07 01:50:35 -0700 | [diff] [blame] | 20 | "conv_s16_b32.cc", |
| 21 | "conv_s16_b64.cc", |
Alex Van Damme | b11ad17 | 2024-04-18 14:18:17 -0700 | [diff] [blame] | 22 | "conv_s8.cc", |
Lun Dong | 3b8d3cb | 2024-05-07 01:50:35 -0700 | [diff] [blame] | 23 | "conv_s8_1x1.cc", |
| 24 | "conv_s8_3x1_d48.cc", |
Alex Van Damme | e617f61 | 2024-05-08 16:41:37 -0700 | [diff] [blame] | 25 | "conv_s8_d4.cc", |
Lun Dong | 3b8d3cb | 2024-05-07 01:50:35 -0700 | [diff] [blame] | 26 | "conv_s8_d32.cc", |
Derek Chow | 86806b3 | 2024-05-09 15:26:01 -0700 | [diff] [blame] | 27 | "conv_s8_d1.cc", |
Derek Chow | 16ffb60 | 2023-10-09 14:27:05 -0700 | [diff] [blame] | 28 | "depthwise_conv_s16.cc", |
Naveen Dodda | be4ab97 | 2024-04-17 17:47:46 +0000 | [diff] [blame] | 29 | "depthwise_conv_s8.cc", |
Alex Van Damme | 59112ad | 2023-08-11 11:10:33 -0700 | [diff] [blame] | 30 | "elementwise_add_s16.cc", |
| 31 | "elementwise_add_s32.cc", |
Alex Van Damme | 7a2eaa4 | 2023-08-17 16:07:18 -0700 | [diff] [blame] | 32 | "elementwise_add_s8.cc", |
| 33 | "leaky_relu_s16.cc", |
| 34 | "leaky_relu_s8.cc", |
Lun Dong | 3b8d3cb | 2024-05-07 01:50:35 -0700 | [diff] [blame] | 35 | "max_pool_s8.cc", |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 36 | "memcpy.cc", |
| 37 | ], |
| 38 | hdrs = [ |
Lun Dong | 3b8d3cb | 2024-05-07 01:50:35 -0700 | [diff] [blame] | 39 | "conv_s8.h", |
| 40 | "conv_util.h", |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 41 | "opt.h", |
Alex Van Damme | 59112ad | 2023-08-11 11:10:33 -0700 | [diff] [blame] | 42 | "util.h", |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 43 | ], |
Alex Van Damme | 7a2eaa4 | 2023-08-17 16:07:18 -0700 | [diff] [blame] | 44 | target_compatible_with = ["@kelvin_sw//platforms/cpu:kelvin"], |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 45 | deps = [ |
Cindy Liu | 43879e4 | 2023-10-18 11:18:03 -0700 | [diff] [blame] | 46 | "//crt", |
Alex Van Damme | f39cadd | 2023-08-28 16:21:20 -0700 | [diff] [blame] | 47 | "@tflite-micro//tensorflow/lite/kernels/internal:common", |
Naveen Dodda | be4ab97 | 2024-04-17 17:47:46 +0000 | [diff] [blame] | 48 | "@tflite-micro//tensorflow/lite/kernels/internal:reference_base", |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 49 | ], |
| 50 | alwayslink = True, |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 51 | ) |