1package( 2 default_visibility = ["//tensorflow:internal"], 3 licenses = ["notice"], 4) 5 6py_library( 7 name = "batching", 8 srcs = ["batching.py"], 9 srcs_version = "PY3", 10 deps = [ 11 "//tensorflow/python:dtypes", 12 "//tensorflow/python:experimental_dataset_ops_gen", 13 "//tensorflow/python:framework_ops", 14 "//tensorflow/python:tensor_shape", 15 "//tensorflow/python:tensor_util", 16 "//tensorflow/python:util", 17 "//tensorflow/python/data/ops:dataset_ops", 18 "//tensorflow/python/data/util:convert", 19 "//tensorflow/python/data/util:nest", 20 "//tensorflow/python/data/util:structure", 21 ], 22) 23 24py_library( 25 name = "cardinality", 26 srcs = ["cardinality.py"], 27 srcs_version = "PY3", 28 deps = [ 29 "//tensorflow/python:experimental_dataset_ops_gen", 30 "//tensorflow/python:util", 31 ], 32) 33 34py_library( 35 name = "compression_ops", 36 srcs = ["compression_ops.py"], 37 srcs_version = "PY3", 38 deps = [ 39 "//tensorflow/python:experimental_dataset_ops_gen", 40 ], 41) 42 43py_library( 44 name = "counter", 45 srcs = ["counter.py"], 46 srcs_version = "PY3", 47 deps = [ 48 ":scan_ops", 49 "//tensorflow/python:dtypes", 50 "//tensorflow/python:framework_ops", 51 "//tensorflow/python:util", 52 "//tensorflow/python/data/ops:dataset_ops", 53 ], 54) 55 56py_library( 57 name = "data_service_ops", 58 srcs = [ 59 "data_service_ops.py", 60 ], 61 srcs_version = "PY3", 62 deps = [ 63 "//tensorflow/python:experimental_dataset_ops_gen", 64 "//tensorflow/python:framework_ops", 65 "//tensorflow/python/data/experimental/service:_pywrap_utils", 66 "//tensorflow/python/data/ops:dataset_ops", 67 "//tensorflow/python/data/ops:options", 68 "//tensorflow/python/data/util:nest", 69 ], 70) 71 72py_library( 73 name = "distribute", 74 srcs = [ 75 "distribute.py", 76 ], 77 srcs_version = "PY3", 78 deps = [ 79 "//tensorflow/python:experimental_dataset_ops_gen", 80 "//tensorflow/python:framework_ops", 81 "//tensorflow/python/data/ops:dataset_ops", 82 "//tensorflow/python/data/ops:options", 83 "//tensorflow/python/data/util:nest", 84 ], 85) 86 87py_library( 88 name = "enumerate_ops", 89 srcs = ["enumerate_ops.py"], 90 srcs_version = "PY3", 91 deps = [ 92 "//tensorflow/python:dtypes", 93 "//tensorflow/python/data/ops:dataset_ops", 94 ], 95) 96 97py_library( 98 name = "error_ops", 99 srcs = ["error_ops.py"], 100 srcs_version = "PY3", 101 deps = [ 102 "//tensorflow/python:experimental_dataset_ops_gen", 103 "//tensorflow/python/data/ops:dataset_ops", 104 "//tensorflow/python/data/util:nest", 105 "//tensorflow/python/data/util:sparse", 106 ], 107) 108 109py_library( 110 name = "get_single_element", 111 srcs = ["get_single_element.py"], 112 srcs_version = "PY3", 113 deps = [ 114 "//tensorflow/python:util", 115 "//tensorflow/python/data/ops:dataset_ops", 116 ], 117) 118 119py_library( 120 name = "grouping", 121 srcs = ["grouping.py"], 122 srcs_version = "PY3", 123 deps = [ 124 "//tensorflow/python:array_ops", 125 "//tensorflow/python:check_ops", 126 "//tensorflow/python:dtypes", 127 "//tensorflow/python:framework_ops", 128 "//tensorflow/python:function", 129 "//tensorflow/python:math_ops", 130 "//tensorflow/python:tensor_shape", 131 "//tensorflow/python:util", 132 "//tensorflow/python/data/ops:dataset_ops", 133 "//tensorflow/python/data/util:nest", 134 "//tensorflow/python/data/util:structure", 135 ], 136) 137 138py_library( 139 name = "interleave_ops", 140 srcs = ["interleave_ops.py"], 141 srcs_version = "PY3", 142 deps = [ 143 ":random_ops", 144 "//tensorflow/python:array_ops", 145 "//tensorflow/python:dtypes", 146 "//tensorflow/python:experimental_dataset_ops_gen", 147 "//tensorflow/python:framework_ops", 148 "//tensorflow/python:math_ops", 149 "//tensorflow/python:stateless_random_ops_gen", 150 "//tensorflow/python:tf2", 151 "//tensorflow/python:util", 152 "//tensorflow/python/data/ops:dataset_ops", 153 "//tensorflow/python/data/ops:readers", 154 "//tensorflow/python/data/util:nest", 155 "//tensorflow/python/data/util:structure", 156 "//tensorflow/python/framework:tensor_spec", 157 "//tensorflow/python/util:tf_export", 158 ], 159) 160 161py_library( 162 name = "io", 163 srcs = [ 164 "io.py", 165 ], 166 srcs_version = "PY3", 167 deps = [ 168 "//tensorflow/python:checkpoint_management", 169 "//tensorflow/python:experimental_dataset_ops_gen", 170 "//tensorflow/python/data/ops:dataset_ops", 171 "//tensorflow/python/training/tracking:util", 172 ], 173) 174 175py_library( 176 name = "iterator_ops", 177 srcs = [ 178 "iterator_ops.py", 179 ], 180 srcs_version = "PY3", 181 deps = [ 182 "//tensorflow/python:basic_session_run_hooks", 183 "//tensorflow/python:checkpoint_management", 184 "//tensorflow/python:dataset_ops_gen", 185 "//tensorflow/python:framework_ops", 186 "//tensorflow/python:saver", 187 "//tensorflow/python:session_run_hook", 188 "//tensorflow/python/data/ops:iterator_ops", 189 "//tensorflow/python/data/ops:optional_ops", 190 ], 191) 192 193py_library( 194 name = "lookup_ops", 195 srcs = [ 196 "lookup_ops.py", 197 ], 198 srcs_version = "PY2AND3", 199 deps = [ 200 ":cardinality", 201 "//tensorflow/python:lookup_ops", 202 ], 203) 204 205py_library( 206 name = "map_defun", 207 srcs = ["map_defun.py"], 208 srcs_version = "PY3", 209 deps = [ 210 "//tensorflow/python:dataset_ops_gen", 211 "//tensorflow/python:framework_ops", 212 "//tensorflow/python:tensor_shape", 213 ], 214) 215 216py_library( 217 name = "matching_files", 218 srcs = ["matching_files.py"], 219 srcs_version = "PY3", 220 deps = [ 221 "//tensorflow/python:dataset_ops_gen", 222 "//tensorflow/python:framework_ops", 223 "//tensorflow/python:tensor_shape", 224 "//tensorflow/python/data/ops:dataset_ops", 225 "//tensorflow/python/data/util:structure", 226 ], 227) 228 229py_library( 230 name = "parsing_ops", 231 srcs = ["parsing_ops.py"], 232 srcs_version = "PY3", 233 deps = [ 234 "//tensorflow/python:dataset_ops_gen", 235 "//tensorflow/python:dtypes", 236 "//tensorflow/python:framework_ops", 237 "//tensorflow/python:parsing_ops", 238 "//tensorflow/python:sparse_tensor", 239 "//tensorflow/python:tensor_shape", 240 "//tensorflow/python/data/ops:dataset_ops", 241 "//tensorflow/python/data/util:structure", 242 ], 243) 244 245py_library( 246 name = "prefetching_ops", 247 srcs = ["prefetching_ops.py"], 248 srcs_version = "PY3", 249 deps = [ 250 "//tensorflow/python:experimental_dataset_ops_gen", 251 "//tensorflow/python:framework_ops", 252 "//tensorflow/python:tensor_spec", 253 "//tensorflow/python/data/ops:dataset_ops", 254 "//tensorflow/python/data/util:nest", 255 "//tensorflow/python/data/util:sparse", 256 "//tensorflow/python/eager:context", 257 "//tensorflow/python/eager:function", 258 ], 259) 260 261py_library( 262 name = "random_access", 263 srcs = ["random_access.py"], 264 srcs_version = "PY3", 265 deps = [ 266 "//tensorflow/python:experimental_dataset_ops_gen", 267 "//tensorflow/python/data/util:structure", 268 ], 269) 270 271py_library( 272 name = "random_ops", 273 srcs = [ 274 "random_ops.py", 275 ], 276 srcs_version = "PY3", 277 deps = [ 278 "//tensorflow/python:constant_op", 279 "//tensorflow/python:framework_ops", 280 "//tensorflow/python:util", 281 "//tensorflow/python/data/ops:dataset_ops", 282 "//tensorflow/python/data/util:structure", 283 ], 284) 285 286py_library( 287 name = "readers", 288 srcs = [ 289 "readers.py", 290 ], 291 srcs_version = "PY3", 292 deps = [ 293 ":batching", 294 ":error_ops", 295 ":interleave_ops", 296 ":parsing_ops", 297 ":shuffle_ops", 298 "//tensorflow/python:constant_op", 299 "//tensorflow/python:dataset_ops_gen", 300 "//tensorflow/python:dtypes", 301 "//tensorflow/python:experimental_dataset_ops_gen", 302 "//tensorflow/python:framework_ops", 303 "//tensorflow/python:io_ops", 304 "//tensorflow/python:lib", 305 "//tensorflow/python:platform", 306 "//tensorflow/python:tensor_shape", 307 "//tensorflow/python:util", 308 "//tensorflow/python/data/ops:dataset_ops", 309 "//tensorflow/python/data/ops:readers", 310 "//tensorflow/python/data/util:convert", 311 "//tensorflow/python/data/util:nest", 312 "//third_party/py/numpy", 313 ], 314) 315 316py_library( 317 name = "resampling", 318 srcs = ["resampling.py"], 319 srcs_version = "PY3", 320 deps = [ 321 "//tensorflow/python:util", 322 "//tensorflow/python/data/ops:dataset_ops", 323 ], 324) 325 326py_library( 327 name = "scan_ops", 328 srcs = ["scan_ops.py"], 329 srcs_version = "PY3", 330 deps = [ 331 "//tensorflow/python:util", 332 ], 333) 334 335py_library( 336 name = "shuffle_ops", 337 srcs = [ 338 "shuffle_ops.py", 339 ], 340 srcs_version = "PY3", 341 deps = [ 342 "//tensorflow/python/data/ops:dataset_ops", 343 ], 344) 345 346py_library( 347 name = "snapshot", 348 srcs = [ 349 "snapshot.py", 350 ], 351 srcs_version = "PY3", 352 visibility = ["//tensorflow:internal"], 353 deps = [ 354 "//tensorflow/python:experimental_dataset_ops_gen", 355 "//tensorflow/python/data/ops:dataset_ops", 356 "//tensorflow/python/data/util:structure", 357 ], 358) 359 360py_library( 361 name = "take_while_ops", 362 srcs = ["take_while_ops.py"], 363 srcs_version = "PY3", 364 deps = [ 365 "//tensorflow/python:util", 366 ], 367) 368 369py_library( 370 name = "testing", 371 srcs = ["testing.py"], 372 srcs_version = "PY3", 373 deps = [ 374 "//tensorflow/python:experimental_dataset_ops_gen", 375 "//tensorflow/python:framework_ops", 376 "//tensorflow/python/data/ops:dataset_ops", 377 ], 378) 379 380py_library( 381 name = "unique", 382 srcs = [ 383 "unique.py", 384 ], 385 srcs_version = "PY3", 386 deps = [ 387 "//tensorflow/python:util", 388 ], 389) 390 391py_library( 392 name = "writers", 393 srcs = [ 394 "writers.py", 395 ], 396 srcs_version = "PY3", 397 deps = [ 398 "//tensorflow/python:dtypes", 399 "//tensorflow/python/data/ops:dataset_ops", 400 ], 401) 402 403py_library( 404 name = "dataset_ops", 405 srcs_version = "PY3", 406 deps = [ 407 ":batching", 408 ":cardinality", 409 ":compression_ops", 410 ":counter", 411 ":data_service_ops", 412 ":distribute", 413 ":enumerate_ops", 414 ":error_ops", 415 ":get_single_element", 416 ":grouping", 417 ":interleave_ops", 418 ":io", 419 ":map_defun", 420 ":matching_files", 421 ":prefetching_ops", 422 ":readers", 423 ":resampling", 424 ":scan_ops", 425 ":shuffle_ops", 426 ":snapshot", 427 ":take_while_ops", 428 ":unique", 429 ":writers", 430 "//tensorflow/python:dataset_ops_gen", 431 "//tensorflow/python:util", 432 "//tensorflow/python/data/ops:dataset_ops", 433 "//tensorflow/python/data/util:nest", 434 ], 435) 436