• 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 = "math_ops",
34    srcs = [
35        "math_ops.cc",
36    ],
37    hdrs = [
38        "math_ops.h",
39    ],
40    visibility = [
41        "//tensorflow:internal",
42    ],
43    deps = [
44        ":array_ops",
45        "//tensorflow/c/eager:abstract_context",
46        "//tensorflow/c/eager:abstract_tensor_handle",
47        "//tensorflow/c/eager:c_api_unified_internal",
48        "//tensorflow/c/eager:tracing_utils",
49        "//tensorflow/core:framework",
50        "//tensorflow/core/platform:errors",
51    ],
52)
53
54cc_library(
55    name = "nn_ops",
56    srcs = [
57        "nn_ops.cc",
58    ],
59    hdrs = [
60        "nn_ops.h",
61    ],
62    visibility = [
63        "//tensorflow:internal",
64    ],
65    deps = [
66        "//tensorflow/c/eager:abstract_context",
67        "//tensorflow/c/eager:abstract_operation",
68        "//tensorflow/c/eager:abstract_tensor_handle",
69        "//tensorflow/c/eager:c_api_unified_internal",
70        "//tensorflow/c/eager:tracing_utils",
71        "//tensorflow/core:framework",
72        "//tensorflow/core/platform:errors",
73    ],
74)
75
76cc_library(
77    name = "resource_variable_ops",
78    srcs = [
79        "resource_variable_ops.cc",
80    ],
81    hdrs = [
82        "resource_variable_ops.h",
83    ],
84    visibility = [
85        "//tensorflow:internal",
86    ],
87    deps = [
88        "//tensorflow/c/eager:abstract_context",
89        "//tensorflow/c/eager:abstract_operation",
90        "//tensorflow/c/eager:abstract_tensor_handle",
91        "//tensorflow/c/eager:c_api_unified_internal",
92        "//tensorflow/c/eager:tracing_utils",
93        "//tensorflow/core:framework",
94        "//tensorflow/core/platform:errors",
95    ],
96)
97
98cc_library(
99    name = "ops",
100    hdrs = [
101        "array_ops.h",
102        "math_ops.h",
103        "nn_ops.h",
104        "resource_variable_ops.h",
105    ],
106    visibility = [
107        "//tensorflow:internal",
108    ],
109    deps = [
110        ":array_ops",
111        ":math_ops",
112        ":nn_ops",
113        ":resource_variable_ops",
114        "//tensorflow/c/eager:abstract_context",
115        "//tensorflow/c/eager:abstract_operation",
116        "//tensorflow/c/eager:abstract_tensor_handle",
117        "//tensorflow/c/eager:c_api_unified_internal",
118    ],
119)
120
121filegroup(
122    name = "pywrap_required_hdrs",
123    srcs = [
124        "array_ops.h",
125        "math_ops.h",
126        "nn_ops.h",
127        "resource_variable_ops.h",
128    ],
129    visibility = [
130        "//tensorflow/core:__pkg__",
131        "//tensorflow/python:__subpackages__",
132    ],
133)
134