A. Unique TensorFlower f635a5bb85 Introduce custom transformations placeholders and rename ApplyModelTransformations.
PiperOrigin-RevId: 329575742
Change-Id: Iefd7c92c7bfdb11ed2e8645ef9e6ad08ddfdd621
2020-09-01 14:01:34 -07:00

26 lines
733 B
Python

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",
],
)
cc_library(
name = "custom_transformations",
srcs = ["custom_transformations.cc"],
hdrs = ["//tensorflow/lite/delegates/gpu/common:custom_transformations.h"],
deps = [
"//tensorflow/lite/delegates/gpu/common:model_transformer",
],
)