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

Merge pull request #124 from bernardohenz/layer-norm 

Adding dependencies for layer normalization
This commit is contained in:
lissyx 2020-08-25 13:18:09 +02:00 committed by GitHub
parent 4336a5b49f 6dc2a1becf
commit 23ad988fcd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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",