Refactor and add GPU kernel

tensorflow · Oct 29, 2020 · fa722c2 · fa722c2
1 parent 480bbb2
commit fa722c2
Show file tree

Hide file tree

Showing 6 changed files with 1,299 additions and 771 deletions.
diff --git a/tensorflow_addons/custom_ops/layers/BUILD b/tensorflow_addons/custom_ops/layers/BUILD
@@ -31,5 +31,7 @@ custom_op_library(
         "@cub_archive//:cub",
     ],
     cuda_srcs = [
+        "cc/kernels/deformable_conv2d_op.h",
+        "cc/kernels/deformable_conv2d_op_gpu.cu.cc",
     ],
 )