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