aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/cmake/tf_models.cmake
blob: 4349f7d09b523e9f5b75be55a63fe720b4473a6d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
########################################################
# tf_models_word2vec_ops library
########################################################
file(GLOB tf_models_word2vec_ops_srcs
    "${tensorflow_source_dir}/tensorflow_models/tutorials/embedding/word2vec_ops.cc"
)

add_library(tf_models_word2vec_ops OBJECT ${tf_models_word2vec_ops_srcs})

add_dependencies(tf_models_word2vec_ops tf_core_framework)

########################################################
# tf_models_word2vec_kernels library
########################################################
file(GLOB tf_models_word2vec_kernels_srcs
    "${tensorflow_source_dir}/tensorflow_models/tutorials/embedding/word2vec_kernels.cc"
)

add_library(tf_models_word2vec_kernels OBJECT ${tf_models_word2vec_kernels_srcs})

add_dependencies(tf_models_word2vec_kernels tf_core_cpu)