1# Copyright 2021 The TensorFlow Authors. All Rights Reserved. 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# ============================================================================== 15 16# C API for delegate plugins. 17 18load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable") 19 20package( 21 default_visibility = ["//visibility:private"], 22 licenses = ["notice"], 23) 24 25cc_library( 26 name = "delegate_plugin", 27 hdrs = ["delegate_plugin.h"], 28 compatible_with = get_compatible_with_portable(), 29 visibility = ["//visibility:public"], 30 deps = [ 31 "//tensorflow/lite/c:common", 32 ], 33) 34 35cc_library( 36 name = "nnapi_plugin", 37 srcs = ["nnapi_plugin.cc"], 38 hdrs = ["nnapi_plugin.h"], 39 visibility = ["//visibility:public"], 40 deps = [ 41 ":delegate_plugin", 42 "//tensorflow/lite/c:common", 43 "//tensorflow/lite/delegates/nnapi:nnapi_delegate", 44 "//tensorflow/lite/experimental/acceleration/configuration:configuration_fbs", 45 "//tensorflow/lite/experimental/acceleration/configuration:nnapi_plugin_impl", 46 ], 47) 48 49cc_library( 50 name = "gpu_plugin", 51 srcs = ["gpu_plugin.cc"], 52 hdrs = ["gpu_plugin.h"], 53 visibility = ["//visibility:public"], 54 deps = [ 55 ":delegate_plugin", 56 "//tensorflow/lite/c:common", 57 "//tensorflow/lite/delegates/gpu:delegate", 58 "//tensorflow/lite/experimental/acceleration/configuration:configuration_fbs", 59 "//tensorflow/lite/experimental/acceleration/configuration:gpu_plugin_impl", 60 ], 61) 62