Skip to content

Commit 23ad988

Browse files
authored
Merge pull request #124 from bernardohenz/layer-norm
Adding dependencies for layer normalization
2 parents 4336a5b + 6dc2a1b commit 23ad988

File tree

1 file changed

+9
-0
lines changed
  • tensorflow/core/kernels

1 file changed

+9
-0
lines changed

tensorflow/core/kernels/BUILD

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8658,11 +8658,20 @@ tf_kernel_library(
86588658
"cwise_op_less.cc",
86598659
"cwise_op_minimum.cc",
86608660
"cwise_op_mul_1.cc",
8661+
"cwise_op_squared_difference.cc",
8662+
"cwise_op_add_1.cc",
8663+
"cwise_op_add_2.cc",
8664+
"cwise_op_rsqrt.cc",
8665+
"cwise_op_sub.cc",
86618666
],
86628667
gpu_srcs = [
86638668
"cwise_op_gpu_less.cu.cc",
86648669
"cwise_op_gpu_minimum.cu.cc",
86658670
"cwise_op_gpu_mul.cu.cc",
8671+
"cwise_op_gpu_squared_difference.cu.cc",
8672+
"cwise_op_gpu_add.cu.cc",
8673+
"cwise_op_gpu_rsqrt.cu.cc",
8674+
"cwise_op_gpu_sub.cu.cc",
86668675
],
86678676
deps = [
86688677
":cwise_lib",

0 commit comments

Comments
 (0)