Employ tf_platform_alias for TFLite GPU custom_parsers.
PiperOrigin-RevId: 306731712 Change-Id: I58bfc7bd622891ff218ef0e1ac4a78aa06d87725
This commit is contained in:
parent
9304bd49ec
commit
98c434cd50
@ -1,3 +1,5 @@
|
||||
load("//tensorflow/core/platform:build_config.bzl", "tf_platform_alias")
|
||||
|
||||
package(
|
||||
default_visibility = ["//visibility:public"],
|
||||
licenses = ["notice"], # Apache 2.0
|
||||
@ -19,17 +21,9 @@ cc_library(
|
||||
],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "custom_parsers",
|
||||
srcs = ["custom_parsers.cc"],
|
||||
hdrs = ["custom_parsers.h"],
|
||||
deps = [
|
||||
":shape",
|
||||
":status",
|
||||
"@com_google_absl//absl/strings",
|
||||
"@com_google_absl//absl/types:any",
|
||||
"@flatbuffers",
|
||||
],
|
||||
exports_files(
|
||||
["custom_parsers.h"],
|
||||
visibility = ["//tensorflow/lite/delegates/gpu/common:__subpackages__"],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
@ -126,7 +120,6 @@ cc_library(
|
||||
srcs = ["model_builder.cc"],
|
||||
hdrs = ["model_builder.h"],
|
||||
deps = [
|
||||
":custom_parsers",
|
||||
":data_type",
|
||||
":model",
|
||||
":model_builder_helper",
|
||||
@ -134,6 +127,10 @@ cc_library(
|
||||
":shape",
|
||||
":status",
|
||||
":tensor",
|
||||
"@FP16",
|
||||
"@com_google_absl//absl/memory",
|
||||
"@com_google_absl//absl/strings",
|
||||
"@com_google_absl//absl/strings:str_format",
|
||||
"//tensorflow/lite:context",
|
||||
"//tensorflow/lite:kernel_api",
|
||||
"//tensorflow/lite:util",
|
||||
@ -143,11 +140,7 @@ cc_library(
|
||||
"//tensorflow/lite/kernels/internal:reference_base",
|
||||
"//tensorflow/lite/kernels/internal:tensor",
|
||||
"//tensorflow/lite/schema:schema_fbs",
|
||||
"@FP16",
|
||||
"@com_google_absl//absl/memory",
|
||||
"@com_google_absl//absl/strings",
|
||||
"@com_google_absl//absl/strings:str_format",
|
||||
],
|
||||
] + tf_platform_alias("custom_parsers", "//tensorflow/lite/delegates/gpu/common/"),
|
||||
)
|
||||
|
||||
cc_test(
|
||||
|
16
tensorflow/lite/delegates/gpu/common/default/BUILD
Normal file
16
tensorflow/lite/delegates/gpu/common/default/BUILD
Normal file
@ -0,0 +1,16 @@
|
||||
package(
|
||||
default_visibility = ["//visibility:public"],
|
||||
licenses = ["notice"], # Apache 2.0
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "custom_parsers",
|
||||
srcs = ["custom_parsers.cc"],
|
||||
hdrs = ["//tensorflow/lite/delegates/gpu/common:custom_parsers.h"],
|
||||
deps = [
|
||||
"//tensorflow/lite/delegates/gpu/common:shape",
|
||||
"//tensorflow/lite/delegates/gpu/common:status",
|
||||
"@com_google_absl//absl/strings",
|
||||
"@com_google_absl//absl/types:any",
|
||||
],
|
||||
)
|
@ -1,4 +1,4 @@
|
||||
/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
|
||||
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
@ -12,6 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#include "tensorflow/lite/delegates/gpu/common/custom_parsers.h"
|
||||
|
||||
#include <string>
|
Loading…
Reference in New Issue
Block a user