diff options
Diffstat (limited to 'tensorflow/examples/label_image')
-rw-r--r-- | tensorflow/examples/label_image/BUILD | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/tensorflow/examples/label_image/BUILD b/tensorflow/examples/label_image/BUILD index 732e5d54f0..545b6ae584 100644 --- a/tensorflow/examples/label_image/BUILD +++ b/tensorflow/examples/label_image/BUILD @@ -7,14 +7,25 @@ licenses(["notice"]) # Apache 2.0 exports_files(["LICENSE"]) +cc_library( + name = "command_line_flags", + srcs = [ + "command_line_flags.cc", + "command_line_flags.h", + ], + deps = [ + "//tensorflow/core:tensorflow", + ], +) + cc_binary( name = "label_image", srcs = [ - "command_line_flags.cc", "main.cc", ], linkopts = ["-lm"], deps = [ + ":command_line_flags", "//tensorflow/cc:cc_ops", "//tensorflow/core:tensorflow", ], @@ -23,11 +34,12 @@ cc_binary( cc_test( name = "command_line_flags_test", srcs = [ - "command_line_flags.cc", "command_line_flags_test.cc", ], deps = [ + ":command_line_flags", "//tensorflow/core:tensorflow", + "//tensorflow/core/platform/default/build_config:gtest", "//tensorflow/core/platform/default/build_config:test_main", ], ) |