1 package( 2 default_visibility = [ 3 "//visibility:public", 4 ], 5 ) 6 7 licenses(["notice"]) # Apache 2.0 8 9 exports_files([ 10 "LICENSE", 11 ]) 12 13 load("//tensorflow:tensorflow.bzl", "tf_cc_binary") 14 load("//tensorflow:tensorflow.bzl", "tf_cc_test") 15 load("//tensorflow:tensorflow.bzl", "tf_py_test") 16 17 py_library( 18 name = "models", 19 srcs = [ 20 "models.py", 21 ], 22 srcs_version = "PY2AND3", 23 deps = [ 24 "//tensorflow:tensorflow_py", 25 "//third_party/py/numpy", 26 "@six_archive//:six", 27 ], 28 ) 29 30 tf_py_test( 31 name = "models_test", 32 size = "small", 33 srcs = ["models_test.py"], 34 additional_deps = [ 35 ":models", 36 "//tensorflow/python:client_testlib", 37 ], 38 ) 39 40 py_library( 41 name = "input_data", 42 srcs = [ 43 "input_data.py", 44 ], 45 srcs_version = "PY2AND3", 46 deps = [ 47 "//tensorflow:tensorflow_py", 48 "//third_party/py/numpy", 49 "@six_archive//:six", 50 ], 51 ) 52 53 tf_py_test( 54 name = "input_data_test", 55 size = "small", 56 srcs = ["input_data_test.py"], 57 additional_deps = [ 58 ":input_data", 59 "//tensorflow/python:client_testlib", 60 ], 61 ) 62 63 py_binary( 64 name = "train", 65 srcs = [ 66 "train.py", 67 ], 68 srcs_version = "PY2AND3", 69 deps = [ 70 ":input_data", 71 ":models", 72 "//tensorflow:tensorflow_py", 73 "//third_party/py/numpy", 74 "@six_archive//:six", 75 ], 76 ) 77 78 py_binary( 79 name = "freeze", 80 srcs = [ 81 "freeze.py", 82 ], 83 srcs_version = "PY2AND3", 84 deps = [ 85 ":input_data", 86 ":models", 87 "//tensorflow:tensorflow_py", 88 "//third_party/py/numpy", 89 "@six_archive//:six", 90 ], 91 ) 92 93 tf_py_test( 94 name = "freeze_test", 95 size = "small", 96 srcs = ["freeze_test.py"], 97 additional_deps = [ 98 ":freeze", 99 "//tensorflow/python:client_testlib", 100 ], 101 ) 102 103 py_binary( 104 name = "generate_streaming_test_wav", 105 srcs = [ 106 "generate_streaming_test_wav.py", 107 ], 108 srcs_version = "PY2AND3", 109 deps = [ 110 ":input_data", 111 ":models", 112 "//tensorflow:tensorflow_py", 113 "//third_party/py/numpy", 114 "@six_archive//:six", 115 ], 116 ) 117 118 tf_py_test( 119 name = "generate_streaming_test_wav_test", 120 size = "small", 121 srcs = ["generate_streaming_test_wav_test.py"], 122 additional_deps = [ 123 ":generate_streaming_test_wav", 124 "//tensorflow/python:client_testlib", 125 ], 126 ) 127 128 tf_cc_binary( 129 name = "label_wav_cc", 130 srcs = [ 131 "label_wav.cc", 132 ], 133 deps = [ 134 "//tensorflow/core:core_cpu", 135 "//tensorflow/core:framework", 136 "//tensorflow/core:framework_internal", 137 "//tensorflow/core:lib", 138 "//tensorflow/core:protos_all_cc", 139 "//tensorflow/core:tensorflow", 140 ], 141 ) 142 143 py_binary( 144 name = "label_wav", 145 srcs = [ 146 "label_wav.py", 147 ], 148 srcs_version = "PY2AND3", 149 deps = [ 150 "//tensorflow:tensorflow_py", 151 ], 152 ) 153 154 tf_py_test( 155 name = "label_wav_test", 156 size = "medium", 157 srcs = ["label_wav_test.py"], 158 additional_deps = [ 159 ":label_wav", 160 "//tensorflow/python:client_testlib", 161 ], 162 ) 163 164 cc_library( 165 name = "recognize_commands", 166 srcs = [ 167 "recognize_commands.cc", 168 ], 169 hdrs = [ 170 "recognize_commands.h", 171 ], 172 deps = [ 173 "//tensorflow/core:core_cpu", 174 "//tensorflow/core:framework", 175 "//tensorflow/core:framework_internal", 176 "//tensorflow/core:lib", 177 "//tensorflow/core:protos_all_cc", 178 "//tensorflow/core:tensorflow", 179 ], 180 ) 181 182 tf_cc_test( 183 name = "recognize_commands_test", 184 size = "medium", 185 srcs = [ 186 "recognize_commands_test.cc", 187 ], 188 deps = [ 189 ":recognize_commands", 190 "//tensorflow/core:lib", 191 "//tensorflow/core:lib_internal", 192 "//tensorflow/core:test", 193 "//tensorflow/core:test_main", 194 "//tensorflow/core:testlib", 195 ], 196 ) 197 198 cc_library( 199 name = "accuracy_utils", 200 srcs = [ 201 "accuracy_utils.cc", 202 ], 203 hdrs = [ 204 "accuracy_utils.h", 205 ], 206 deps = [ 207 "//tensorflow/core:core_cpu", 208 "//tensorflow/core:framework", 209 "//tensorflow/core:framework_internal", 210 "//tensorflow/core:lib", 211 "//tensorflow/core:protos_all_cc", 212 "//tensorflow/core:tensorflow", 213 ], 214 ) 215 216 tf_cc_test( 217 name = "accuracy_utils_test", 218 size = "medium", 219 srcs = [ 220 "accuracy_utils_test.cc", 221 ], 222 deps = [ 223 ":accuracy_utils", 224 "//tensorflow/core:lib", 225 "//tensorflow/core:lib_internal", 226 "//tensorflow/core:test", 227 "//tensorflow/core:test_main", 228 "//tensorflow/core:testlib", 229 ], 230 ) 231 232 tf_cc_binary( 233 name = "test_streaming_accuracy", 234 srcs = [ 235 "test_streaming_accuracy.cc", 236 ], 237 deps = [ 238 ":accuracy_utils", 239 ":recognize_commands", 240 "//tensorflow/core:core_cpu", 241 "//tensorflow/core:framework", 242 "//tensorflow/core:framework_internal", 243 "//tensorflow/core:lib", 244 "//tensorflow/core:lib_internal", 245 "//tensorflow/core:protos_all_cc", 246 ], 247 ) 248 249 filegroup( 250 name = "all_files", 251 srcs = glob( 252 ["**/*"], 253 exclude = [ 254 "**/METADATA", 255 "**/OWNERS", 256 ], 257 ), 258 visibility = ["//tensorflow:__subpackages__"], 259 ) 260
