Add sigmoid and tanh to deepspeech_cwise_ops

pull/4/head
Reuben Morais 2 years ago
parent 23ad988fcd
commit ca5d9fdf5c
  1. 14
      tensorflow/core/kernels/BUILD

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

Loading…
Cancel
Save