• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2
3package(
4    default_visibility = [
5        ":friends",
6    ],
7    licenses = ["notice"],  # Apache 2.0
8)
9
10package_group(
11    name = "friends",
12    includes = ["//third_party/mlir:subpackages"],
13    packages = [
14        "//learning/brain/experimental/mlir/...",
15        "//tensorflow/lite/...",
16    ],
17)
18
19cc_library(
20    name = "sparsify_model",
21    srcs = [
22        "sparsify_model.cc",
23    ],
24    hdrs = [
25        "sparsify_model.h",
26    ],
27    deps = [
28        "//tensorflow/compiler/mlir/lite:common",
29        "//tensorflow/compiler/mlir/lite:flatbuffer_translate_lib",
30        "//tensorflow/compiler/mlir/lite:tensorflow_lite_d2s",
31        "//tensorflow/compiler/mlir/tensorflow:error_util",
32        "//tensorflow/core:protos_all_cc",
33        "//tensorflow/lite:framework",
34        "//tensorflow/lite/core/api",
35        "//tensorflow/lite/schema:schema_fbs",
36        "@com_google_absl//absl/strings",
37        "@llvm-project//llvm:Support",
38        "@llvm-project//mlir:IR",
39        "@llvm-project//mlir:Pass",
40    ],
41)
42