• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2
3licenses(["notice"])  # Apache 2.0
4
5package(default_visibility = [":friends"])
6
7package_group(
8    name = "friends",
9    packages = [
10        "//tensorflow/lite/toco/...",
11    ],
12)
13
14cc_library(
15    name = "tf_tfl_flatbuffer_helpers",
16    srcs = ["tf_tfl_flatbuffer_helpers.cc"],
17    hdrs = ["tf_tfl_flatbuffer_helpers.h"],
18    deps = [
19        "//tensorflow/compiler/mlir/lite:common",
20        "//tensorflow/compiler/mlir/lite:tensorflow_lite",
21        "//tensorflow/compiler/mlir/lite:tf_tfl_passes",
22        "//tensorflow/compiler/mlir/lite:tf_to_tfl_flatbuffer",
23        "//tensorflow/compiler/mlir/tensorflow",
24        "//tensorflow/compiler/mlir/tensorflow:convert_graphdef",
25        "//tensorflow/compiler/mlir/tensorflow:dump_mlir_util",
26        "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags",
27        "//tensorflow/core:core_cpu_base",
28        "//tensorflow/core:lib",
29        "//tensorflow/core:protos_all_cc",
30        "//tensorflow/lite/toco:model_flags_proto_cc",
31        "//tensorflow/lite/toco:toco_flags_proto_cc",
32        "//tensorflow/lite/toco:types_proto_cc",
33        "//tensorflow/stream_executor/lib",
34        "@llvm-project//llvm:Support",
35        "@llvm-project//mlir:IR",
36        "@llvm-project//mlir:Pass",
37        "@llvm-project//mlir:Support",
38        "@llvm-project//mlir:Transforms",
39    ],
40)
41
42cc_library(
43    name = "graphdef_to_tfl_flatbuffer",
44    srcs = ["graphdef_to_tfl_flatbuffer.cc"],
45    hdrs = [
46        "graphdef_to_tfl_flatbuffer.h",
47    ],
48    deps = [
49        ":tf_tfl_flatbuffer_helpers",
50        "//tensorflow/compiler/mlir/lite:common",
51        "//tensorflow/compiler/mlir/lite:tensorflow_lite",
52        "//tensorflow/compiler/mlir/lite:tf_tfl_passes",
53        "//tensorflow/compiler/mlir/lite:tf_to_tfl_flatbuffer",
54        "//tensorflow/compiler/mlir/tensorflow",
55        "//tensorflow/compiler/mlir/tensorflow:convert_graphdef",
56        "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags",
57        "//tensorflow/core:lib",
58        "//tensorflow/core:protos_all_cc",
59        "//tensorflow/lite/toco:model_flags_proto_cc",
60        "//tensorflow/lite/toco:toco_flags_proto_cc",
61        "//tensorflow/lite/toco:types_proto_cc",
62        "//tensorflow/stream_executor/lib",
63        "@llvm-project//llvm:Support",
64        "@llvm-project//mlir:IR",
65        "@llvm-project//mlir:Pass",
66        "@llvm-project//mlir:Support",
67        "@llvm-project//mlir:Transforms",
68    ],
69)
70
71cc_library(
72    name = "saved_model_to_tfl_flatbuffer",
73    srcs = ["saved_model_to_tfl_flatbuffer.cc"],
74    hdrs = [
75        "saved_model_to_tfl_flatbuffer.h",
76    ],
77    deps = [
78        ":tf_tfl_flatbuffer_helpers",
79        "//tensorflow/compiler/mlir/lite:common",
80        "//tensorflow/compiler/mlir/lite:tensorflow_lite",
81        "//tensorflow/compiler/mlir/lite:tf_tfl_passes",
82        "//tensorflow/compiler/mlir/lite:tf_to_tfl_flatbuffer",
83        "//tensorflow/compiler/mlir/tensorflow",
84        "//tensorflow/compiler/mlir/tensorflow:convert_graphdef",
85        "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags",
86        "//tensorflow/core:lib",
87        "//tensorflow/core:protos_all_cc",
88        "//tensorflow/lite/toco:model_flags_proto_cc",
89        "//tensorflow/lite/toco:toco_flags_proto_cc",
90        "//tensorflow/lite/toco:types_proto_cc",
91        "//tensorflow/stream_executor/lib",
92        "@com_google_absl//absl/types:span",
93        "@llvm-project//llvm:Support",
94        "@llvm-project//mlir:IR",
95        "@llvm-project//mlir:Pass",
96        "@llvm-project//mlir:Support",
97        "@llvm-project//mlir:Transforms",
98    ],
99)
100