• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Experimental C++ SavedModel Header Only APIs. See RFC
2# https://github.com/tensorflow/community/pull/207
3
4load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
5
6package(
7    # This is intentionally public
8    default_visibility = [
9        "//visibility:public",
10    ],
11    licenses = ["notice"],
12)
13
14cc_library(
15    name = "concrete_function",
16    hdrs = [
17        "concrete_function.h",
18    ],
19    deps = [
20        ":function_metadata",
21        "//tensorflow/c/eager:c_api",
22        "//tensorflow/c/experimental/saved_model/public:concrete_function",
23        "//tensorflow/cc/experimental/base/public:status",
24    ],
25)
26
27cc_library(
28    name = "concrete_function_list",
29    hdrs = [
30        "concrete_function_list.h",
31    ],
32    deps = [
33        ":concrete_function",
34        "//tensorflow/c/experimental/saved_model/public:concrete_function_list",
35    ],
36)
37
38cc_library(
39    name = "function_metadata",
40    hdrs = [
41        "function_metadata.h",
42    ],
43    deps = [
44        "//tensorflow/c/experimental/saved_model/public:function_metadata",
45    ],
46)
47
48cc_library(
49    name = "saved_model_api",
50    hdrs = [
51        "saved_model_api.h",
52    ],
53    deps = [
54        ":concrete_function",
55        ":concrete_function_list",
56        ":signature_def_function",
57        "//tensorflow/c/experimental/saved_model/public:saved_model_api",
58        "//tensorflow/cc/experimental/base/public:runtime",
59        "//tensorflow/cc/experimental/base/public:status",
60    ],
61)
62
63cc_library(
64    name = "signature_def_function",
65    hdrs = [
66        "signature_def_function.h",
67    ],
68    deps = [
69        ":signature_def_function_metadata",
70        "//tensorflow/c/eager:c_api",
71        "//tensorflow/c/experimental/saved_model/public:signature_def_function",
72        "//tensorflow/cc/experimental/base/public:status",
73    ],
74)
75
76cc_library(
77    name = "signature_def_function_metadata",
78    hdrs = [
79        "signature_def_function_metadata.h",
80    ],
81    deps = [
82        "//tensorflow/c/experimental/saved_model/public:signature_def_function_metadata",
83    ],
84)
85