Skip to content

Commit c350d0c

Browse files
committed
Create kernel library for Less, Minimum, and Mul ops
1 parent 3dc6405 commit c350d0c

File tree

1 file changed

+20
-0
lines changed
  • tensorflow/core/kernels

1 file changed

+20
-0
lines changed

tensorflow/core/kernels/BUILD

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7673,6 +7673,26 @@ cc_library(
76737673
],
76747674
)
76757675

7676+
tf_kernel_library(
7677+
name = "deepspeech_cwise_ops",
7678+
srcs = [
7679+
"cwise_op_less.cc",
7680+
"cwise_op_minimum.cc",
7681+
"cwise_op_mul_1.cc",
7682+
],
7683+
gpu_srcs = [
7684+
"cwise_op_gpu_less.cu.cc",
7685+
"cwise_op_gpu_minimum.cu.cc",
7686+
"cwise_op_gpu_mul.cu.cc",
7687+
],
7688+
deps = [
7689+
":cwise_lib",
7690+
"//tensorflow/core:framework",
7691+
"//tensorflow/core:lib",
7692+
"//third_party/eigen3",
7693+
],
7694+
)
7695+
76767696
# Header-only version of cwise_lib for clients that want to use the cwise_ops
76777697
# functionality in their own custom ops.
76787698
cc_header_only_library(

0 commit comments

Comments
 (0)