• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description:
2#   C++ implementation code for the summary writing APIs.
3
4load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
5load(
6    "//tensorflow:tensorflow.bzl",
7    "tf_cc_binary",
8    "tf_cc_test",
9    "tf_copts",
10)
11
12package(
13    default_visibility = ["//tensorflow:internal"],
14    licenses = ["notice"],
15)
16
17cc_library(
18    name = "schema",
19    srcs = ["schema.cc"],
20    hdrs = ["schema.h"],
21    copts = tf_copts(),
22    deps = [
23        "//tensorflow/core:lib",
24        "//tensorflow/core/lib/db:sqlite",
25    ],
26)
27
28tf_cc_test(
29    name = "schema_test",
30    size = "small",
31    srcs = ["schema_test.cc"],
32    deps = [
33        ":schema",
34        "//tensorflow/core:test",
35        "//tensorflow/core:test_main",
36    ],
37)
38
39cc_library(
40    name = "summary_db_writer",
41    srcs = ["summary_db_writer.cc"],
42    hdrs = ["summary_db_writer.h"],
43    copts = tf_copts(),
44    deps = [
45        ":summary_converter",
46        "//tensorflow/core:framework",
47        "//tensorflow/core:lib",
48        "//tensorflow/core:lib_internal",
49        "//tensorflow/core:protos_all_cc",
50        "//tensorflow/core/kernels:summary_interface",
51        "//tensorflow/core/lib/db:sqlite",
52    ],
53)
54
55tf_cc_test(
56    name = "summary_db_writer_test",
57    size = "small",
58    srcs = ["summary_db_writer_test.cc"],
59    deps = [
60        ":schema",
61        ":summary_db_writer",
62        "//tensorflow/core:lib",
63        "//tensorflow/core:protos_all_cc",
64        "//tensorflow/core:test",
65        "//tensorflow/core:test_main",
66        "//tensorflow/core/lib/db:sqlite",
67    ],
68)
69
70cc_library(
71    name = "summary_file_writer",
72    srcs = ["summary_file_writer.cc"],
73    hdrs = ["summary_file_writer.h"],
74    copts = tf_copts(),
75    deps = [
76        ":summary_converter",
77        "//tensorflow/core:framework",
78        "//tensorflow/core:lib",
79        "//tensorflow/core:lib_internal",
80        "//tensorflow/core:protos_all_cc",
81        "//tensorflow/core:ptr_util",
82        "//tensorflow/core/kernels:summary_interface",
83        "@com_google_absl//absl/strings",
84    ],
85)
86
87tf_cc_test(
88    name = "summary_file_writer_test",
89    size = "medium",  # file i/o
90    timeout = "short",
91    srcs = ["summary_file_writer_test.cc"],
92    deps = [
93        ":summary_file_writer",
94        "//tensorflow/core:framework",
95        "//tensorflow/core:lib",
96        "//tensorflow/core:lib_internal",
97        "//tensorflow/core:protos_all_cc",
98        "//tensorflow/core:test",
99        "//tensorflow/core:test_main",
100        "@com_google_absl//absl/strings",
101    ],
102)
103
104cc_library(
105    name = "summary_converter",
106    srcs = ["summary_converter.cc"],
107    hdrs = ["summary_converter.h"],
108    copts = tf_copts(),
109    visibility = ["//visibility:private"],
110    deps = [
111        "//tensorflow/core:framework",
112        "//tensorflow/core:lib",
113        "//tensorflow/core:lib_internal",
114        "//tensorflow/core:protos_all_cc",
115        "//tensorflow/core/lib/png:png_io",
116    ],
117)
118
119tf_cc_binary(
120    name = "loader",
121    srcs = ["loader.cc"],
122    linkstatic = 1,
123    deps = [
124        ":schema",
125        ":summary_db_writer",
126        "//tensorflow/core:framework",
127        "//tensorflow/core:framework_internal",
128        "//tensorflow/core:lib",
129        "//tensorflow/core:protos_all_cc",
130        "//tensorflow/core/lib/db:sqlite",
131    ],
132)
133
134tf_cc_binary(
135    name = "vacuum",
136    srcs = ["vacuum.cc"],
137    linkstatic = 1,
138    deps = [
139        "//tensorflow/core:framework_internal",
140        "//tensorflow/core:lib",
141        "//tensorflow/core/lib/db:sqlite",
142    ],
143)
144