Create kernel library for Less, Minimum, and Mul ops

mozilla · Jul 8, 2019 · c350d0c · c350d0c
1 parent 3dc6405
commit c350d0c
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
@@ -7673,6 +7673,26 @@ cc_library(
     ],
 )
 
+tf_kernel_library(
+    name = "deepspeech_cwise_ops",
+    srcs = [
+        "cwise_op_less.cc",
+        "cwise_op_minimum.cc",
+        "cwise_op_mul_1.cc",
+    ],
+    gpu_srcs = [
+        "cwise_op_gpu_less.cu.cc",
+        "cwise_op_gpu_minimum.cu.cc",
+        "cwise_op_gpu_mul.cu.cc",
+    ],
+    deps = [
+        ":cwise_lib",
+        "//tensorflow/core:framework",
+        "//tensorflow/core:lib",
+        "//third_party/eigen3",
+    ],
+)
+
 # Header-only version of cwise_lib for clients that want to use the cwise_ops
 # functionality in their own custom ops.
 cc_header_only_library(