From 6dc2a1becfd1316eb4d77240133a548e93dbff63 Mon Sep 17 00:00:00 2001 From: Bernardo Henz Date: Wed, 19 Aug 2020 13:05:58 -0300 Subject: [PATCH] Adding dependencies for layer normalization --- tensorflow/core/kernels/BUILD | 9 +++++++++ 1 file changed, 9 insertions(+) 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",