We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cc89403 commit f540233Copy full SHA for f540233
tensorflow_addons/custom_ops/layers/BUILD
@@ -10,14 +10,13 @@ cc_binary(
10
srcs = [
11
"cc/kernels/correlation_cost_op.cc",
12
"cc/kernels/correlation_cost_op.h",
13
- "cc/kernels/correlation_cost_op_gpu.cu.cc",
14
"cc/ops/correlation_cost_op.cc",
15
],
16
copts = [
17
"-pthread",
18
"-std=c++11",
19
D_GLIBCXX_USE_CXX11_ABI,
20
- ],
+ ] + if_cuda(["-DGOOGLE_CUDA=1"]),
21
linkshared = 1,
22
deps = [
23
"@local_config_tf//:libtensorflow_framework",
0 commit comments