13
13
"if_libtpu" ,
14
14
_tsl_clean_dep = "clean_dep" ,
15
15
)
16
- load (
17
- "//tensorflow:tensorflow.bzl" ,
18
- "clean_dep" ,
19
- )
20
16
load ("@local_config_cuda//cuda:build_defs.bzl" , "if_cuda" )
21
17
load ("@local_config_rocm//rocm:build_defs.bzl" , "if_rocm" )
22
18
load (
@@ -32,34 +28,34 @@ def tf_dtensor_tpu_dependencies():
32
28
33
29
def tf_additional_binary_deps ():
34
30
return [
35
- clean_dep ( "@nsync//:nsync_cpp" ),
31
+ str ( Label ( "@nsync//:nsync_cpp" ) ),
36
32
# TODO(allenl): Split these out into their own shared objects. They are
37
33
# here because they are shared between contrib/ op shared objects and
38
34
# core.
39
- clean_dep ( "//tensorflow/core/kernels:lookup_util" ),
40
- clean_dep ( "//tensorflow/core/util/tensor_bundle" ),
35
+ str ( Label ( "//tensorflow/core/kernels:lookup_util" ) ),
36
+ str ( Label ( "//tensorflow/core/util/tensor_bundle" ) ),
41
37
] + if_cuda (
42
38
[
43
- clean_dep ( "@local_xla//xla/stream_executor:cuda_platform" ),
39
+ str ( Label ( "@local_xla//xla/stream_executor:cuda_platform" ) ),
44
40
],
45
41
) + if_rocm (
46
42
[
47
- clean_dep ( "@local_xla//xla/stream_executor:rocm_platform" ),
48
- clean_dep ( "@local_xla//xla/stream_executor/rocm:rocm_rpath" ),
43
+ str ( Label ( "@local_xla//xla/stream_executor:rocm_platform" ) ),
44
+ str ( Label ( "@local_xla//xla/stream_executor/rocm:rocm_rpath" ) ),
49
45
],
50
46
) + if_mkl_ml (
51
47
[
52
- clean_dep ( "//third_party/mkl:intel_binary_blob" ),
48
+ str ( Label ( "//third_party/mkl:intel_binary_blob" ) ),
53
49
],
54
50
)
55
51
56
52
def tf_protos_all ():
57
53
return if_static (
58
54
extra_deps = [
59
- clean_dep ( "//tensorflow/core/protobuf:conv_autotuning_proto_cc_impl" ),
60
- clean_dep ( "//tensorflow/core:protos_all_cc_impl" ),
55
+ str ( Label ( "//tensorflow/core/protobuf:conv_autotuning_proto_cc_impl" ) ),
56
+ str ( Label ( "//tensorflow/core:protos_all_cc_impl" ) ),
61
57
_xla_clean_dep ("@local_xla//xla:autotuning_proto_cc_impl" ),
62
58
_tsl_clean_dep ("@local_tsl//tsl/protobuf:protos_all_cc_impl" ),
63
59
],
64
- otherwise = [clean_dep ( "//tensorflow/core:protos_all_cc" )],
60
+ otherwise = [str ( Label ( "//tensorflow/core:protos_all_cc" ) )],
65
61
)
0 commit comments