1 package(default_visibility = ["//visibility:public"]) 2 3 licenses(["notice"]) # Apache 2.0 4 5 load( 6 "//tensorflow:tensorflow.bzl", 7 "tf_cc_test", 8 ) 9 10 cc_library( 11 name = "cluster_utils", 12 srcs = [ 13 "cluster_utils.cc", 14 ], 15 hdrs = [ 16 "cluster_utils.h", 17 ], 18 deps = [ 19 "//tensorflow/contrib/lite/toco:toco_port", 20 ], 21 ) 22 23 cc_library( 24 name = "cluster", 25 srcs = [ 26 "cluster.cc", 27 ], 28 hdrs = [ 29 "cluster.h", 30 ], 31 deps = [ 32 ":cluster_utils", 33 "//tensorflow/contrib/lite/toco:model", 34 "//tensorflow/contrib/lite/toco:tooling_util", 35 "//tensorflow/core:protos_all_cc", 36 ], 37 ) 38 39 cc_library( 40 name = "resolve_svdf", 41 srcs = [ 42 "resolve_svdf.cc", 43 ], 44 hdrs = [ 45 "resolve_svdf.h", 46 ], 47 visibility = ["//visibility:public"], 48 deps = [ 49 ":cluster", 50 ":cluster_utils", 51 "//tensorflow/contrib/lite/toco:model", 52 "//tensorflow/contrib/lite/toco:toco_port", 53 "//tensorflow/contrib/lite/toco:tooling_util", 54 "//tensorflow/core:lib", 55 "//tensorflow/core:protos_all_cc", 56 "@protobuf_archive//:protobuf_headers", 57 ], 58 ) 59 60 tf_cc_test( 61 name = "resolve_svdf_test", 62 srcs = ["resolve_svdf_test.cc"], 63 deps = [ 64 ":cluster", 65 ":cluster_utils", 66 ":resolve_cluster", 67 ":resolve_svdf", 68 "//tensorflow/core:lib", 69 "//tensorflow/core:protos_all_cc", 70 "@com_google_googletest//:gtest_main", 71 ], 72 ) 73 74 cc_library( 75 name = "resolve_cluster", 76 srcs = [ 77 "resolve_cluster.cc", 78 ], 79 hdrs = [ 80 "resolve_cluster.h", 81 ], 82 visibility = ["//visibility:public"], 83 deps = [ 84 ":cluster", 85 ":cluster_utils", 86 ":resolve_svdf", 87 "//tensorflow/contrib/lite/toco:tooling_util", 88 "//tensorflow/core:protos_all_cc", 89 ], 90 ) 91 92 filegroup( 93 name = "all_files", 94 srcs = glob( 95 ["**/*"], 96 exclude = [ 97 "**/METADATA", 98 "**/OWNERS", 99 ], 100 ), 101 visibility = ["//tensorflow:__subpackages__"], 102 ) 103