diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 8546525e0dd..6e435830f86 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -8658,11 +8658,20 @@ tf_kernel_library( "cwise_op_less.cc", "cwise_op_minimum.cc", "cwise_op_mul_1.cc", + "cwise_op_squared_difference.cc", + "cwise_op_add_1.cc", + "cwise_op_add_2.cc", + "cwise_op_rsqrt.cc", + "cwise_op_sub.cc", ], gpu_srcs = [ "cwise_op_gpu_less.cu.cc", "cwise_op_gpu_minimum.cu.cc", "cwise_op_gpu_mul.cu.cc", + "cwise_op_gpu_squared_difference.cu.cc", + "cwise_op_gpu_add.cu.cc", + "cwise_op_gpu_rsqrt.cu.cc", + "cwise_op_gpu_sub.cu.cc", ], deps = [ ":cwise_lib",