From ca5d9fdf5c0a9ecae2c650c5222ac690a0a6f966 Mon Sep 17 00:00:00 2001 From: Reuben Morais Date: Sat, 2 Jan 2021 17:17:29 +0000 Subject: [PATCH] Add sigmoid and tanh to deepspeech_cwise_ops --- tensorflow/core/kernels/BUILD | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 6e435830f86..14f4d2c040d 100644 --- a/tensorflow/core/kernels/BUILD +++ b/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",