tensorflow
diff --git a/‎tensorflow_addons/custom_ops/layers/BUILD‎
Lines changed: 2 additions & 0 deletions b/‎tensorflow_addons/custom_ops/layers/BUILD‎
Lines changed: 2 additions & 0 deletions
@@ -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",
     ],
 )
Original file line number	Diff line number	Diff line change
`@@ -31,5 +31,7 @@ custom_op_library(`
`31`	`31`	`"@cub_archive//:cub",`
`32`	`32`	`],`
`33`	`33`	`cuda_srcs = [`
	`34`	`+ "cc/kernels/deformable_conv2d_op.h",`
	`35`	`+ "cc/kernels/deformable_conv2d_op_gpu.cu.cc",`
`34`	`36`	`],`
`35`	`37`	`)`