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 = [ |
Alex Van Damme | f39cadd | 2023-08-28 16:21:20 -0700 | [diff] [blame] | 20 | "conv.cc", |
Derek Chow | 16ffb60 | 2023-10-09 14:27:05 -0700 | [diff] [blame] | 21 | "depthwise_conv_s16.cc", |
Alex Van Damme | 59112ad | 2023-08-11 11:10:33 -0700 | [diff] [blame] | 22 | "elementwise_add_s16.cc", |
| 23 | "elementwise_add_s32.cc", |
Alex Van Damme | 7a2eaa4 | 2023-08-17 16:07:18 -0700 | [diff] [blame] | 24 | "elementwise_add_s8.cc", |
| 25 | "leaky_relu_s16.cc", |
| 26 | "leaky_relu_s8.cc", |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 27 | "memcpy.cc", |
Naveen-Dodda | 3285c6a | 2023-12-07 20:14:06 +0000 | [diff] [blame] | 28 | "max_pool_s8.cc" |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 29 | ], |
| 30 | hdrs = [ |
| 31 | "opt.h", |
Alex Van Damme | 59112ad | 2023-08-11 11:10:33 -0700 | [diff] [blame] | 32 | "util.h", |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 33 | ], |
Alex Van Damme | 7a2eaa4 | 2023-08-17 16:07:18 -0700 | [diff] [blame] | 34 | target_compatible_with = ["@kelvin_sw//platforms/cpu:kelvin"], |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 35 | deps = [ |
Cindy Liu | 43879e4 | 2023-10-18 11:18:03 -0700 | [diff] [blame] | 36 | "//crt", |
Alex Van Damme | f39cadd | 2023-08-28 16:21:20 -0700 | [diff] [blame] | 37 | "@tflite-micro//tensorflow/lite/kernels/internal:common", |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 38 | ], |
| 39 | alwayslink = True, |
Alex Van Damme | d555403 | 2023-08-01 15:39:38 -0700 | [diff] [blame] | 40 | ) |