@@ -11,38 +11,68 @@ cc_library(
11
11
name = "box_util" ,
12
12
srcs = ["box_util.cc" ],
13
13
hdrs = ["box_util.h" ],
14
+ copts = select ({
15
+ ":windows" : [
16
+ "/DEIGEN_STRONG_INLINE=inline" ,
17
+ "-DTENSORFLOW_MONOLITHIC_BUILD" ,
18
+ "/DPLATFORM_WINDOWS" ,
19
+ "/DEIGEN_HAS_C99_MATH" ,
20
+ "/DTENSORFLOW_USE_EIGEN_THREADPOOL" ,
21
+ "/DEIGEN_AVOID_STL_ARRAY" ,
22
+ "/Iexternal/gemmlowp" ,
23
+ "/wd4018" ,
24
+ "/wd4577" ,
25
+ "/DNOGDI" ,
26
+ "/UTF_COMPILE_LIBRARY" ,
27
+ ],
28
+ "//conditions:default" : [
29
+ "-pthread" ,
30
+ "-std=c++17" ,
31
+ ],
32
+ }),
14
33
deps = [
15
34
"@local_config_tf//:libtensorflow_framework" ,
16
35
"@local_config_tf//:tf_header_lib" ,
17
36
],
18
- copts = select ({
19
- ":windows" : ["/DEIGEN_STRONG_INLINE=inline" , "-DTENSORFLOW_MONOLITHIC_BUILD" , "/DPLATFORM_WINDOWS" , "/DEIGEN_HAS_C99_MATH" , "/DTENSORFLOW_USE_EIGEN_THREADPOOL" , "/DEIGEN_AVOID_STL_ARRAY" , "/Iexternal/gemmlowp" , "/wd4018" , "/wd4577" , "/DNOGDI" , "/UTF_COMPILE_LIBRARY" ],
20
- "//conditions:default" : ["-pthread" , "-std=c++17" ],
21
- }),
22
37
)
23
38
24
39
cc_binary (
25
- name = ' _keras_cv_custom_ops.so' ,
40
+ name = " _keras_cv_custom_ops.so" ,
26
41
srcs = [
27
42
"kernels/pairwise_iou_kernel.cc" ,
28
- "ops/pairwise_iou_op.cc" ,
29
- "kernels/withinbox_op.cc" ,
30
- "ops/withinbox_op.cc" ,
31
43
"kernels/within_any_box_op.cc" ,
44
+ "kernels/withinbox_op.cc" ,
45
+ "ops/pairwise_iou_op.cc" ,
32
46
"ops/within_any_box_op.cc" ,
47
+ "ops/withinbox_op.cc" ,
33
48
],
49
+ copts = select ({
50
+ ":windows" : [
51
+ "/DEIGEN_STRONG_INLINE=inline" ,
52
+ "-DTENSORFLOW_MONOLITHIC_BUILD" ,
53
+ "/DPLATFORM_WINDOWS" ,
54
+ "/DEIGEN_HAS_C99_MATH" ,
55
+ "/DTENSORFLOW_USE_EIGEN_THREADPOOL" ,
56
+ "/DEIGEN_AVOID_STL_ARRAY" ,
57
+ "/Iexternal/gemmlowp" ,
58
+ "/wd4018" ,
59
+ "/wd4577" ,
60
+ "/DNOGDI" ,
61
+ "/UTF_COMPILE_LIBRARY" ,
62
+ ],
63
+ "//conditions:default" : [
64
+ "-pthread" ,
65
+ "-std=c++17" ,
66
+ ],
67
+ }),
68
+ features = select ({
69
+ ":windows" : ["windows_export_all_symbols" ],
70
+ "//conditions:default" : [],
71
+ }),
34
72
linkshared = 1 ,
35
73
deps = [
74
+ ":box_util" ,
36
75
"@local_config_tf//:libtensorflow_framework" ,
37
76
"@local_config_tf//:tf_header_lib" ,
38
- ":box_util" ,
39
77
],
40
- features = select ({
41
- ":windows" : ["windows_export_all_symbols" ],
42
- "//conditions:default" : [],
43
- }),
44
- copts = select ({
45
- ":windows" : ["/DEIGEN_STRONG_INLINE=inline" , "-DTENSORFLOW_MONOLITHIC_BUILD" , "/DPLATFORM_WINDOWS" , "/DEIGEN_HAS_C99_MATH" , "/DTENSORFLOW_USE_EIGEN_THREADPOOL" , "/DEIGEN_AVOID_STL_ARRAY" , "/Iexternal/gemmlowp" , "/wd4018" , "/wd4577" , "/DNOGDI" , "/UTF_COMPILE_LIBRARY" ],
46
- "//conditions:default" : ["-pthread" , "-std=c++17" ],
47
- }),
48
78
)
0 commit comments