blob: 14b6b17e6106e7a74c02340637922482c798fd09 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
set(tf_op_lib_names
"array_ops"
"candidate_sampling_ops"
"control_flow_ops"
"ctc_ops"
"data_flow_ops"
"functional_ops"
"image_ops"
"io_ops"
"linalg_ops"
"logging_ops"
"losses"
"math_ops"
"nn_ops"
"no_op"
"parsing_ops"
"random_ops"
"resource_variable_ops"
"script_ops"
"sdca_ops"
"set_ops"
"sendrecv_ops"
"sparse_ops"
"state_ops"
"string_ops"
"training_ops"
)
foreach(tf_op_lib_name ${tf_op_lib_names})
########################################################
# tf_${tf_op_lib_name} library
########################################################
file(GLOB tf_${tf_op_lib_name}_srcs
"${tensorflow_source_dir}/tensorflow/core/ops/${tf_op_lib_name}.cc"
)
add_library(tf_${tf_op_lib_name} OBJECT ${tf_${tf_op_lib_name}_srcs})
add_dependencies(tf_${tf_op_lib_name} tf_core_framework)
endforeach()
function(GENERATE_CONTRIB_OP_LIBRARY op_lib_name cc_srcs)
add_library(tf_contrib_${op_lib_name}_ops OBJECT ${cc_srcs})
add_dependencies(tf_contrib_${op_lib_name}_ops tf_core_framework)
endfunction()
GENERATE_CONTRIB_OP_LIBRARY(cudnn_rnn "${tensorflow_source_dir}/tensorflow/contrib/cudnn_rnn/ops/cudnn_rnn_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(factorization_clustering "${tensorflow_source_dir}/tensorflow/contrib/factorization/ops/clustering_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(factorization_factorization "${tensorflow_source_dir}/tensorflow/contrib/factorization/ops/factorization_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(framework_variable "${tensorflow_source_dir}/tensorflow/contrib/framework/ops/variable_ops.cc")
GENERATE_CONTRIB_OP_LIBRARY(tensor_forest "${tensorflow_source_dir}/tensorflow/contrib/tensor_forest/ops/tensor_forest_ops.cc")
########################################################
# tf_user_ops library
########################################################
file(GLOB_RECURSE tf_user_ops_srcs
"${tensorflow_source_dir}/tensorflow/core/user_ops/*.cc"
)
add_library(tf_user_ops OBJECT ${tf_user_ops_srcs})
add_dependencies(tf_user_ops tf_core_framework)
########################################################
# tf_core_ops library
########################################################
file(GLOB_RECURSE tf_core_ops_srcs
"${tensorflow_source_dir}/tensorflow/core/ops/*.h"
"${tensorflow_source_dir}/tensorflow/core/ops/*.cc"
"${tensorflow_source_dir}/tensorflow/core/user_ops/*.h"
"${tensorflow_source_dir}/tensorflow/core/user_ops/*.cc"
)
file(GLOB_RECURSE tf_core_ops_exclude_srcs
"${tensorflow_source_dir}/tensorflow/core/ops/*test*.h"
"${tensorflow_source_dir}/tensorflow/core/ops/*test*.cc"
"${tensorflow_source_dir}/tensorflow/core/ops/*main.cc"
"${tensorflow_source_dir}/tensorflow/core/user_ops/*test*.h"
"${tensorflow_source_dir}/tensorflow/core/user_ops/*test*.cc"
"${tensorflow_source_dir}/tensorflow/core/user_ops/*main.cc"
"${tensorflow_source_dir}/tensorflow/core/user_ops/*.cu.cc"
)
list(REMOVE_ITEM tf_core_ops_srcs ${tf_core_ops_exclude_srcs})
add_library(tf_core_ops OBJECT ${tf_core_ops_srcs})
add_dependencies(tf_core_ops tf_core_cpu)
|