Skip to content

Commit 467c1c2

Browse files
gunantensorflower-gardener
gunan
authored and
tensorflower-gardener
committed
Add direct dependencies to tf op libraries.
PiperOrigin-RevId: 230034058
1 parent fdd4979 commit 467c1c2

File tree

11 files changed

+74
-0
lines changed

11 files changed

+74
-0
lines changed

‎tensorflow/c/BUILD

+7
Original file line numberDiff line numberDiff line change
@@ -289,13 +289,20 @@ tf_cuda_cc_test(
289289
"//tensorflow/cc/saved_model:signature_constants",
290290
"//tensorflow/cc/saved_model:tag_constants",
291291
"//tensorflow/compiler/jit",
292+
"//tensorflow/core:array_ops_op_lib",
293+
"//tensorflow/core:bitwise_ops_op_lib",
294+
"//tensorflow/core:control_flow_ops_op_lib",
292295
"//tensorflow/core:core_cpu_internal",
293296
"//tensorflow/core:direct_session",
294297
"//tensorflow/core:framework",
295298
"//tensorflow/core:framework_internal",
299+
"//tensorflow/core:functional_ops_op_lib",
296300
"//tensorflow/core:lib",
301+
"//tensorflow/core:math_ops_op_lib",
302+
"//tensorflow/core:nn_ops_op_lib",
297303
"//tensorflow/core:proto_text",
298304
"//tensorflow/core:protos_all_cc",
305+
"//tensorflow/core:spectral_ops_op_lib",
299306
"//tensorflow/core:test",
300307
"//tensorflow/core:test_main",
301308
"//tensorflow/core/kernels:array",

‎tensorflow/compiler/jit/BUILD

+6
Original file line numberDiff line numberDiff line change
@@ -175,12 +175,18 @@ cc_library(
175175
"//tensorflow/compiler/xla/client:global_data",
176176
"//tensorflow/compiler/xla/client:local_client",
177177
"//tensorflow/compiler/xla/service:stream_pool",
178+
"//tensorflow/core:array_ops_op_lib",
179+
"//tensorflow/core:control_flow_ops_op_lib",
178180
"//tensorflow/core:core_cpu",
179181
"//tensorflow/core:core_cpu_internal",
180182
"//tensorflow/core:framework",
183+
"//tensorflow/core:functional_ops_op_lib",
181184
"//tensorflow/core:lib",
182185
"//tensorflow/core:lib_internal",
186+
"//tensorflow/core:math_ops_op_lib",
187+
"//tensorflow/core:nn_ops_op_lib",
183188
"//tensorflow/core:protos_all_cc",
189+
"//tensorflow/core:resource_variable_ops_op_lib",
184190
"//tensorflow/core:stream_executor_no_cuda",
185191
"//tensorflow/core/kernels:cast_op",
186192
"//tensorflow/core/kernels:constant_op",

‎tensorflow/compiler/tf2xla/kernels/BUILD

+9
Original file line numberDiff line numberDiff line change
@@ -144,13 +144,22 @@ tf_kernel_library(
144144
"//tensorflow/compiler/xla/client/lib:quantize",
145145
"//tensorflow/compiler/xla/client/lib:sorting",
146146
"//tensorflow/compiler/xla/client/lib:triangular_solve",
147+
"//tensorflow/core:bitwise_ops_op_lib",
148+
"//tensorflow/core:control_flow_ops_op_lib",
147149
"//tensorflow/core:framework",
150+
"//tensorflow/core:functional_ops_op_lib",
148151
"//tensorflow/core:image_ops_op_lib",
149152
"//tensorflow/core:lib",
150153
"//tensorflow/core:linalg_ops_op_lib",
154+
"//tensorflow/core:list_ops_op_lib",
155+
"//tensorflow/core:math_ops_op_lib",
156+
"//tensorflow/core:nn_ops_op_lib",
151157
"//tensorflow/core:protos_all_cc",
158+
"//tensorflow/core:random_ops_op_lib",
159+
"//tensorflow/core:resource_variable_ops_op_lib",
152160
"//tensorflow/core:spectral_ops_op_lib",
153161
"//tensorflow/core:stateless_random_ops_op_lib",
162+
"//tensorflow/core:training_ops_op_lib",
154163
"//tensorflow/core/kernels:bounds_check",
155164
"//tensorflow/core/kernels:concat_lib",
156165
"//tensorflow/core/kernels:constant_op",

‎tensorflow/core/common_runtime/eager/BUILD

+4
Original file line numberDiff line numberDiff line change
@@ -181,11 +181,15 @@ tf_cc_test(
181181
"//tensorflow/cc:client_session",
182182
"//tensorflow/cc:ops",
183183
"//tensorflow/cc:scope",
184+
"//tensorflow/core:array_ops_op_lib",
184185
"//tensorflow/core:core_cpu",
185186
"//tensorflow/core:core_cpu_internal",
186187
"//tensorflow/core:direct_session",
187188
"//tensorflow/core:framework",
189+
"//tensorflow/core:functional_ops_op_lib",
188190
"//tensorflow/core:lib",
191+
"//tensorflow/core:math_ops_op_lib",
192+
"//tensorflow/core:nn_ops_op_lib",
189193
"//tensorflow/core:protos_all_cc",
190194
"//tensorflow/core:test",
191195
"//tensorflow/core:test_main",

‎tensorflow/core/debug/BUILD

+4
Original file line numberDiff line numberDiff line change
@@ -221,10 +221,14 @@ tf_cc_test(
221221
deps = [
222222
":debug_grpc_testlib",
223223
":debug_io_utils",
224+
"//tensorflow/core:array_ops_op_lib",
224225
"//tensorflow/core:core_cpu",
225226
"//tensorflow/core:framework",
227+
"//tensorflow/core:functional_ops_op_lib",
226228
"//tensorflow/core:lib",
227229
"//tensorflow/core:master_proto_cc",
230+
"//tensorflow/core:math_ops_op_lib",
231+
"//tensorflow/core:nn_ops_op_lib",
228232
"//tensorflow/core:protos_all_cc",
229233
"//tensorflow/core:test",
230234
"//tensorflow/core:test_main",

‎tensorflow/core/distributed_runtime/BUILD

+6
Original file line numberDiff line numberDiff line change
@@ -648,6 +648,7 @@ tf_cuda_cc_test(
648648
":remote_device",
649649
":worker_interface",
650650
"//tensorflow:grpc++",
651+
"//tensorflow/core:control_flow_ops_op_lib",
651652
"//tensorflow/core:core_cpu",
652653
"//tensorflow/core:core_cpu_internal",
653654
"//tensorflow/core:framework",
@@ -713,9 +714,14 @@ tf_cuda_cc_test(
713714
tags = tf_cuda_tests_tags(),
714715
deps = [
715716
"//tensorflow/cc:cc_ops",
717+
"//tensorflow/core:array_ops_op_lib",
718+
"//tensorflow/core:bitwise_ops_op_lib",
716719
"//tensorflow/core:core_cpu",
717720
"//tensorflow/core:framework",
721+
"//tensorflow/core:functional_ops_op_lib",
718722
"//tensorflow/core:lib",
723+
"//tensorflow/core:math_ops_op_lib",
724+
"//tensorflow/core:nn_ops_op_lib",
719725
"//tensorflow/core:protos_all_cc",
720726
"//tensorflow/core:tensorflow",
721727
"//tensorflow/core:test",

‎tensorflow/core/distributed_runtime/rpc/BUILD

+5
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,9 @@ tf_cc_binary(
314314
"//tensorflow:grpc++",
315315
"//tensorflow/core:core_cpu",
316316
"//tensorflow/core:framework_internal",
317+
"//tensorflow/core:functional_ops_op_lib",
317318
"//tensorflow/core:lib",
319+
"//tensorflow/core:math_ops_op_lib",
318320
"//tensorflow/core:protos_all_cc",
319321
"//tensorflow/core/distributed_runtime:server_lib",
320322
"//tensorflow/core/kernels:data_flow",
@@ -330,9 +332,12 @@ tf_cc_binary(
330332
deps = [
331333
":grpc_server_lib",
332334
"//tensorflow:grpc++",
335+
"//tensorflow/core:array_ops_op_lib",
336+
"//tensorflow/core:bitwise_ops_op_lib",
333337
"//tensorflow/core:core_cpu",
334338
"//tensorflow/core:framework_internal",
335339
"//tensorflow/core:lib",
340+
"//tensorflow/core:nn_ops_op_lib",
336341
"//tensorflow/core:protos_all_cc",
337342
"//tensorflow/core:testlib",
338343
"//tensorflow/core/distributed_runtime:server_lib",

‎tensorflow/core/kernels/data/BUILD

+7
Original file line numberDiff line numberDiff line change
@@ -76,13 +76,18 @@ tf_cc_test(
7676
srcs = ["single_threaded_executor_test.cc"],
7777
deps = [
7878
":single_threaded_executor",
79+
"//tensorflow/core:bitwise_ops_op_lib",
80+
"//tensorflow/core:control_flow_ops_op_lib",
7981
"//tensorflow/core:core_cpu",
8082
"//tensorflow/core:core_cpu_internal",
8183
"//tensorflow/core:framework",
8284
"//tensorflow/core:framework_internal",
8385
"//tensorflow/core:lib",
8486
"//tensorflow/core:lib_internal",
87+
"//tensorflow/core:math_ops_op_lib",
8588
"//tensorflow/core:protos_all_cc",
89+
"//tensorflow/core:random_ops_op_lib",
90+
"//tensorflow/core:spectral_ops_op_lib",
8691
"//tensorflow/core:test",
8792
"//tensorflow/core:test_main",
8893
"//tensorflow/core:testlib",
@@ -566,6 +571,8 @@ tf_kernel_library(
566571
":tensor_slice_dataset_op",
567572
":window_dataset_op",
568573
":zip_dataset_op",
574+
"//tensorflow/core:array_ops_op_lib",
575+
"//tensorflow/core:nn_ops_op_lib",
569576
"//tensorflow/core/kernels/data/experimental:dataset_kernels",
570577
],
571578
)

‎tensorflow/core/kernels/data/experimental/BUILD

+5
Original file line numberDiff line numberDiff line change
@@ -119,12 +119,14 @@ tf_kernel_library(
119119
name = "map_and_batch_dataset_op",
120120
srcs = ["map_and_batch_dataset_op.cc"],
121121
deps = [
122+
"//tensorflow/core:array_ops_op_lib",
122123
"//tensorflow/core:core_cpu_internal",
123124
"//tensorflow/core:dataset_ops_op_lib",
124125
"//tensorflow/core:framework",
125126
"//tensorflow/core:lib",
126127
"//tensorflow/core:lib_internal",
127128
"//tensorflow/core:metrics",
129+
"//tensorflow/core:nn_ops_op_lib",
128130
"//tensorflow/core/kernels:inplace_ops",
129131
"//tensorflow/core/kernels/data:captured_function",
130132
"//tensorflow/core/kernels/data:dataset_utils",
@@ -169,11 +171,13 @@ tf_kernel_library(
169171
name = "numa_map_and_batch_dataset_op",
170172
srcs = ["numa_map_and_batch_dataset_op.cc"],
171173
deps = [
174+
"//tensorflow/core:array_ops_op_lib",
172175
"//tensorflow/core:core_cpu_internal",
173176
"//tensorflow/core:experimental_dataset_ops_op_lib",
174177
"//tensorflow/core:framework",
175178
"//tensorflow/core:lib",
176179
"//tensorflow/core:lib_internal",
180+
"//tensorflow/core:nn_ops_op_lib",
177181
"//tensorflow/core/kernels:inplace_ops",
178182
"//tensorflow/core/kernels/data:captured_function",
179183
"@com_google_absl//absl/memory",
@@ -200,6 +204,7 @@ tf_kernel_library(
200204
deps = [
201205
"//tensorflow/core:core_cpu_internal",
202206
"//tensorflow/core:framework",
207+
"//tensorflow/core:functional_ops_op_lib",
203208
"//tensorflow/core/kernels/data:parallel_map_iterator",
204209
],
205210
)

‎tensorflow/core/kernels/hexagon/BUILD

+7
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,18 @@ tf_cc_test(
2424
deps = [
2525
":graph_transferer",
2626
"//tensorflow/cc:cc_ops",
27+
"//tensorflow/core:array_ops_op_lib",
28+
"//tensorflow/core:bitwise_ops_op_lib",
2729
"//tensorflow/core:core_cpu",
2830
"//tensorflow/core:direct_session",
31+
"//tensorflow/core:functional_ops_op_lib",
2932
"//tensorflow/core:lib",
33+
"//tensorflow/core:math_ops_op_lib",
34+
"//tensorflow/core:mkl_nn_ops_op_lib",
35+
"//tensorflow/core:nn_ops_op_lib",
3036
"//tensorflow/core:ops",
3137
"//tensorflow/core:protos_all_cc",
38+
"//tensorflow/core:remote_fused_graph_ops_op_lib",
3239
"//tensorflow/core:test",
3340
"//tensorflow/core:test_main",
3441
"//tensorflow/core:testlib",

‎tensorflow/tools/graph_transforms/BUILD

+14
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,19 @@ cc_library(
136136
] + if_not_windows([
137137
"//tensorflow/core/kernels:remote_fused_graph_rewriter_transform",
138138
"//tensorflow/core/kernels/hexagon:hexagon_rewriter_transform",
139+
"//tensorflow/core:user_ops_op_lib",
140+
"//tensorflow/core:training_ops_op_lib",
141+
"//tensorflow/core:string_ops_op_lib",
142+
"//tensorflow/core:remote_fused_graph_ops_op_lib",
143+
"//tensorflow/core:random_ops_op_lib",
144+
"//tensorflow/core:nn_ops_op_lib",
145+
"//tensorflow/core:math_ops_op_lib",
146+
"//tensorflow/core:manip_ops_op_lib",
147+
"//tensorflow/core:list_ops_op_lib",
148+
"//tensorflow/core:functional_ops_op_lib",
149+
"//tensorflow/core:control_flow_ops_op_lib",
150+
"//tensorflow/core:candidate_sampling_ops_op_lib",
151+
"//tensorflow/core:array_ops_op_lib",
139152
]) + if_not_v2([
140153
"//tensorflow/contrib/rnn:gru_ops_op_lib",
141154
"//tensorflow/contrib/rnn:lstm_ops_op_lib",
@@ -175,6 +188,7 @@ tf_cc_test(
175188
":transforms_lib",
176189
"//tensorflow/cc:cc_ops",
177190
"//tensorflow/cc:sendrecv_ops",
191+
"//tensorflow/core:bitwise_ops_op_lib",
178192
"//tensorflow/core:core_cpu",
179193
"//tensorflow/core:lib",
180194
"//tensorflow/core:protos_all_cc",

0 commit comments

Comments
 (0)