• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow:tensorflow.bzl", "filegroup")
2load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
3
4# Experimental ops. These will eventually be replaced by machine-generated versions.
5package(
6    licenses = ["notice"],
7)
8
9cc_library(
10    name = "array_ops",
11    srcs = [
12        "array_ops.cc",
13    ],
14    hdrs = [
15        "array_ops.h",
16    ],
17    visibility = [
18        "//tensorflow:internal",
19    ],
20    deps = [
21        "//tensorflow/c/eager:abstract_context",
22        "//tensorflow/c/eager:abstract_operation",
23        "//tensorflow/c/eager:abstract_tensor_handle",
24        "//tensorflow/c/eager:c_api_unified_internal",
25        "//tensorflow/c/eager:tracing_utils",
26        "//tensorflow/core:framework",
27        "//tensorflow/core/platform:errors",
28        "@com_google_absl//absl/types:span",
29    ],
30)
31
32cc_library(
33    name = "io_ops",
34    srcs = [
35        "io_ops.cc",
36    ],
37    hdrs = [
38        "io_ops.h",
39    ],
40    visibility = [
41        "//tensorflow:internal",
42    ],
43    deps = [
44        "//tensorflow/c/eager:abstract_context",
45        "//tensorflow/c/eager:abstract_tensor_handle",
46        "//tensorflow/c/eager:tracing_utils",
47        "//tensorflow/core:framework",
48        "//tensorflow/core/platform:errors",
49    ],
50)
51
52cc_library(
53    name = "math_ops",
54    srcs = [
55        "math_ops.cc",
56    ],
57    hdrs = [
58        "math_ops.h",
59    ],
60    visibility = [
61        "//tensorflow:internal",
62    ],
63    deps = [
64        ":array_ops",
65        "//tensorflow/c/eager:abstract_context",
66        "//tensorflow/c/eager:abstract_tensor_handle",
67        "//tensorflow/c/eager:c_api_unified_internal",
68        "//tensorflow/c/eager:tracing_utils",
69        "//tensorflow/core:framework",
70        "//tensorflow/core/platform:errors",
71    ],
72)
73
74cc_library(
75    name = "nn_ops",
76    srcs = [
77        "nn_ops.cc",
78    ],
79    hdrs = [
80        "nn_ops.h",
81    ],
82    visibility = [
83        "//tensorflow:internal",
84    ],
85    deps = [
86        "//tensorflow/c/eager:abstract_context",
87        "//tensorflow/c/eager:abstract_operation",
88        "//tensorflow/c/eager:abstract_tensor_handle",
89        "//tensorflow/c/eager:c_api_unified_internal",
90        "//tensorflow/c/eager:tracing_utils",
91        "//tensorflow/core:framework",
92        "//tensorflow/core/platform:errors",
93    ],
94)
95
96cc_library(
97    name = "resource_variable_ops",
98    srcs = [
99        "resource_variable_ops.cc",
100    ],
101    hdrs = [
102        "resource_variable_ops.h",
103    ],
104    visibility = [
105        "//tensorflow:internal",
106    ],
107    deps = [
108        "//tensorflow/c/eager:abstract_context",
109        "//tensorflow/c/eager:abstract_operation",
110        "//tensorflow/c/eager:abstract_tensor_handle",
111        "//tensorflow/c/eager:c_api_unified_internal",
112        "//tensorflow/c/eager:tracing_utils",
113        "//tensorflow/core:framework",
114        "//tensorflow/core/platform:errors",
115    ],
116)
117
118cc_library(
119    name = "ops",
120    hdrs = [
121        "array_ops.h",
122        "io_ops.h",
123        "math_ops.h",
124        "nn_ops.h",
125        "resource_variable_ops.h",
126    ],
127    visibility = [
128        "//tensorflow:internal",
129    ],
130    deps = [
131        ":array_ops",
132        ":io_ops",
133        ":math_ops",
134        ":nn_ops",
135        ":resource_variable_ops",
136        "//tensorflow/c/eager:abstract_context",
137        "//tensorflow/c/eager:abstract_operation",
138        "//tensorflow/c/eager:abstract_tensor_handle",
139        "//tensorflow/c/eager:c_api_unified_internal",
140    ],
141)
142
143filegroup(
144    name = "pywrap_required_hdrs",
145    srcs = [
146        "array_ops.h",
147        "io_ops.h",
148        "math_ops.h",
149        "nn_ops.h",
150        "resource_variable_ops.h",
151    ],
152    visibility = ["//tensorflow/python:__subpackages__"],
153)
154