diff options
| -rw-r--r-- | tensorflow/core/kernels/BUILD | 43 |
1 files changed, 30 insertions, 13 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index ab69925d04..96ccc06f9e 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -4049,11 +4049,6 @@ cc_library( ) SPARSE_DEPS = [ - ":bounds_check", - ":cwise_op", - ":fill_functor", - ":scatter_functor", - "//third_party/eigen3", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:sparse_ops_op_lib", @@ -4086,7 +4081,9 @@ tf_kernel_library( tf_kernel_library( name = "sparse_cross_op", prefix = "sparse_cross_op", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + "//third_party/eigen3", + ], ) tf_kernel_library( @@ -4098,13 +4095,19 @@ tf_kernel_library( tf_kernel_library( name = "sparse_dense_binary_op_shared", prefix = "sparse_dense_binary_op_shared", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + ":cwise_op", + "//third_party/eigen3", + ], ) tf_kernel_library( name = "sparse_sparse_binary_op_shared", prefix = "sparse_sparse_binary_op_shared", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + ":cwise_op", + "//third_party/eigen3", + ], ) tf_kernel_library( @@ -4136,7 +4139,9 @@ tf_kernel_library( tf_kernel_library( name = "sparse_softmax", prefix = "sparse_softmax", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + "//third_party/eigen3", + ], ) tf_kernel_library( @@ -4148,25 +4153,37 @@ tf_kernel_library( tf_kernel_library( name = "sparse_tensor_dense_add_op", prefix = "sparse_tensor_dense_add_op", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + ":scatter_functor", + "//third_party/eigen3", + ], ) tf_kernel_library( name = "sparse_tensor_dense_matmul_op", prefix = "sparse_tensor_dense_matmul_op", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + ":bounds_check", + ":fill_functor", + "//third_party/eigen3", + ], ) tf_kernel_library( name = "sparse_to_dense_op", prefix = "sparse_to_dense_op", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + "//third_party/eigen3", + ], ) tf_kernel_library( name = "sparse_xent_op", prefix = "sparse_xent_op", - deps = SPARSE_DEPS, + deps = SPARSE_DEPS + [ + ":bounds_check", + "//third_party/eigen3", + ], ) tf_kernel_library( |