Fix buildifier errors
This commit is contained in:
parent
d25e90e5cf
commit
0e0c2c734c
@ -524,7 +524,7 @@ cc_library(
|
||||
"@cub_archive//:cub",
|
||||
]) + if_rocm([
|
||||
"@local_config_rocm//rocm:rocprim",
|
||||
])
|
||||
]),
|
||||
)
|
||||
|
||||
cc_library(
|
||||
@ -1393,9 +1393,9 @@ tf_kernel_library(
|
||||
"where_op_gpu_impl_8.cu.cc",
|
||||
],
|
||||
deps = if_cuda_or_rocm([
|
||||
":cuda_solvers",
|
||||
]) + [":gpu_prim_hdrs",]
|
||||
+ ARRAY_DEPS,
|
||||
":cuda_solvers",
|
||||
]) + [":gpu_prim_hdrs"] +
|
||||
ARRAY_DEPS,
|
||||
)
|
||||
|
||||
cc_library(
|
||||
@ -3078,8 +3078,7 @@ tf_kernel_library(
|
||||
tf_kernel_library(
|
||||
name = "generate_box_proposals_op",
|
||||
gpu_srcs = ["generate_box_proposals_op.cu.cc"],
|
||||
deps = [":gpu_prim_hdrs"]
|
||||
+ if_cuda([
|
||||
deps = [":gpu_prim_hdrs"] + if_cuda([
|
||||
":non_max_suppression_op_gpu",
|
||||
]),
|
||||
)
|
||||
@ -4162,9 +4161,10 @@ tf_kernel_library(
|
||||
name = "reduction_ops",
|
||||
gpu_srcs = ["reduction_gpu_kernels.cu.h"],
|
||||
prefix = "reduction_ops",
|
||||
deps = MATH_DEPS +
|
||||
[":gpu_prim_hdrs",
|
||||
":transpose_functor"],
|
||||
deps = MATH_DEPS + [
|
||||
":gpu_prim_hdrs",
|
||||
":transpose_functor",
|
||||
],
|
||||
)
|
||||
|
||||
tf_kernel_library(
|
||||
@ -4856,20 +4856,21 @@ tf_kernel_library(
|
||||
"//tensorflow/core:lib",
|
||||
"//tensorflow/core:lib_internal",
|
||||
"//third_party/eigen3",
|
||||
],
|
||||
],
|
||||
)
|
||||
|
||||
tf_kernel_library(
|
||||
name = "l2loss_op",
|
||||
prefix = "l2loss_op",
|
||||
deps = [
|
||||
":reduction_ops",
|
||||
":gpu_prim_hdrs",
|
||||
"//third_party/eigen3",
|
||||
":reduction_ops",
|
||||
"//tensorflow/core:framework",
|
||||
"//tensorflow/core:lib",
|
||||
"//tensorflow/core:lib_internal",
|
||||
"//tensorflow/core:nn_grad"],
|
||||
"//tensorflow/core:nn_grad",
|
||||
"//third_party/eigen3",
|
||||
],
|
||||
)
|
||||
|
||||
tf_cuda_cc_test(
|
||||
|
Loading…
Reference in New Issue
Block a user