experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Adding dependencies for layer normalization

This commit is contained in:
Bernardo Henz 2020-08-19 13:05:58 -03:00
parent 4336a5b49f
commit 6dc2a1becf
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/core/kernels/BUILD
View File

@ -8658,11 +8658,20 @@ tf_kernel_library(
"cwise_op_less.cc", "cwise_op_less.cc",
"cwise_op_minimum.cc", "cwise_op_minimum.cc",
"cwise_op_mul_1.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 = [ gpu_srcs = [
"cwise_op_gpu_less.cu.cc", "cwise_op_gpu_less.cu.cc",
"cwise_op_gpu_minimum.cu.cc", "cwise_op_gpu_minimum.cu.cc",
"cwise_op_gpu_mul.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 = [ deps = [
":cwise_lib", ":cwise_lib",