diff options
Diffstat (limited to 'tensorflow/contrib/cmake/tf_tools.cmake')
-rw-r--r-- | tensorflow/contrib/cmake/tf_tools.cmake | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/tf_tools.cmake b/tensorflow/contrib/cmake/tf_tools.cmake index 2aaa9ed53e..5151fdb444 100644 --- a/tensorflow/contrib/cmake/tf_tools.cmake +++ b/tensorflow/contrib/cmake/tf_tools.cmake @@ -106,3 +106,22 @@ target_link_libraries(${compare_graphs} PUBLIC ${tf_core_gpu_kernels_lib} ${tensorflow_EXTERNAL_LIBRARIES} ) + +set(benchmark_model "benchmark_model") + +add_executable(${benchmark_model} + "${tensorflow_source_dir}/tensorflow/tools/benchmark/benchmark_model.cc" + "${tensorflow_source_dir}/tensorflow/tools/benchmark/benchmark_model_main.cc" + $<TARGET_OBJECTS:tf_core_lib> + $<TARGET_OBJECTS:tf_core_cpu> + $<TARGET_OBJECTS:tf_core_framework> + $<TARGET_OBJECTS:tf_core_ops> + $<TARGET_OBJECTS:tf_core_direct_session> + $<TARGET_OBJECTS:tf_core_kernels> +) + +target_link_libraries(${benchmark_model} PUBLIC + tf_protos_cc + ${tf_core_gpu_kernels_lib} + ${tensorflow_EXTERNAL_LIBRARIES} +) |