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

Add sigmoid and tanh to deepspeech_cwise_ops

This commit is contained in:
Reuben Morais 2021-01-02 17:17:29 +00:00
parent 23ad988fcd
commit ca5d9fdf5c
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/core/kernels/BUILD
View File

@ -8655,23 +8655,27 @@ cc_library(
tf_kernel_library( tf_kernel_library(
name = "deepspeech_cwise_ops", name = "deepspeech_cwise_ops",
srcs = [ srcs = [
"cwise_op_add_1.cc",
"cwise_op_add_2.cc",
"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_rsqrt.cc",
"cwise_op_squared_difference.cc",
"cwise_op_sub.cc", "cwise_op_sub.cc",
"cwise_op_sigmoid.cc",
"cwise_op_tanh.cc",
], ],
gpu_srcs = [ gpu_srcs = [
"cwise_op_gpu_add.cu.cc",
"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_rsqrt.cu.cc",
"cwise_op_gpu_squared_difference.cu.cc",
"cwise_op_gpu_sub.cu.cc", "cwise_op_gpu_sub.cu.cc",
"cwise_op_gpu_sigmoid.cu.cc",
"cwise_op_gpu_tanh.cu.cc",
], ],
deps = [ deps = [
":cwise_lib", ":cwise_lib",