1# Tests of TensorFlow kernels written using the Python API. 2 3package( 4 default_visibility = ["//tensorflow:internal"], 5) 6 7licenses(["notice"]) # Apache 2.0 8 9load("//tensorflow:tensorflow.bzl", "tf_py_test") 10load("//tensorflow:tensorflow.bzl", "cuda_py_test") 11load("//tensorflow:tensorflow.bzl", "sycl_py_test") 12 13# CPU only tests should use tf_py_test, GPU tests use cuda_py_test 14# Please avoid the py_tests and cuda_py_tests (plural) while we 15# fix the shared/overbroad dependencies. 16 17tf_py_test( 18 name = "as_string_op_test", 19 size = "small", 20 srcs = ["as_string_op_test.py"], 21 additional_deps = [ 22 "//third_party/py/numpy", 23 "//tensorflow/python:array_ops", 24 "//tensorflow/python:client_testlib", 25 "//tensorflow/python:framework_for_generated_wrappers", 26 "//tensorflow/python:string_ops", 27 ], 28 tags = ["no_windows"], 29) 30 31tf_py_test( 32 name = "attention_ops_test", 33 size = "small", 34 srcs = ["attention_ops_test.py"], 35 additional_deps = [ 36 "//third_party/py/numpy", 37 "//tensorflow/python:array_ops", 38 "//tensorflow/python:client_testlib", 39 "//tensorflow/python:framework_for_generated_wrappers", 40 "//tensorflow/python:image_ops", 41 ], 42) 43 44tf_py_test( 45 name = "barrier_ops_test", 46 size = "medium", # NOTE(ebrevdo): This test is NOT small. 47 srcs = ["barrier_ops_test.py"], 48 additional_deps = [ 49 "//third_party/py/numpy", 50 "//tensorflow/python:client_testlib", 51 "//tensorflow/python:data_flow_ops", 52 "//tensorflow/python:errors", 53 "//tensorflow/python:framework_for_generated_wrappers", 54 ], 55 shard_count = 20, 56) 57 58tf_py_test( 59 name = "base64_ops_test", 60 size = "small", 61 srcs = ["base64_ops_test.py"], 62 additional_deps = [ 63 "//third_party/py/numpy", 64 "//tensorflow/python:array_ops", 65 "//tensorflow/python:client_testlib", 66 "//tensorflow/python:errors", 67 "//tensorflow/python:framework_for_generated_wrappers", 68 "//tensorflow/python:framework_test_lib", 69 "//tensorflow/python:string_ops", 70 ], 71 tags = ["nomac"], # b/35468214 72) 73 74tf_py_test( 75 name = "bcast_ops_test", 76 size = "small", 77 srcs = ["bcast_ops_test.py"], 78 additional_deps = [ 79 "//tensorflow/python:array_ops_gen", 80 "//tensorflow/python:client_testlib", 81 ], 82) 83 84cuda_py_test( 85 name = "list_ops_test", 86 size = "small", 87 srcs = ["list_ops_test.py"], 88 additional_deps = [ 89 "//third_party/py/numpy", 90 "//tensorflow/python:array_ops", 91 "//tensorflow/python:math_ops", 92 "//tensorflow/python:list_ops", 93 "//tensorflow/python/eager:context", 94 "//tensorflow/python:framework_for_generated_wrappers", 95 "//tensorflow/python:framework_test_lib", 96 "//tensorflow/python:client_testlib", 97 ], 98 grpc_enabled = True, 99) 100 101cuda_py_test( 102 name = "benchmark_test", 103 size = "small", 104 srcs = ["benchmark_test.py"], 105 additional_deps = [ 106 "//tensorflow/python:client", 107 "//tensorflow/python:client_testlib", 108 "//tensorflow/python:framework_for_generated_wrappers", 109 "//tensorflow/python:platform", 110 "//tensorflow/python:platform_benchmark", 111 ], 112 tags = ["no_windows"], 113) 114 115tf_py_test( 116 name = "bincount_op_test", 117 size = "small", 118 srcs = ["bincount_op_test.py"], 119 additional_deps = [ 120 "//tensorflow/python:client_testlib", 121 "//tensorflow/python:math_ops", 122 "//tensorflow/python:framework_for_generated_wrappers", 123 ], 124) 125 126tf_py_test( 127 name = "candidate_sampler_ops_test", 128 size = "small", 129 srcs = ["candidate_sampler_ops_test.py"], 130 additional_deps = [ 131 "//third_party/py/numpy", 132 "//tensorflow/python:array_ops", 133 "//tensorflow/python:candidate_sampling_ops", 134 "//tensorflow/python:client_testlib", 135 "//tensorflow/python:framework_for_generated_wrappers", 136 "//tensorflow/python:math_ops", 137 ], 138) 139 140tf_py_test( 141 name = "checkpoint_ops_test", 142 size = "medium", 143 srcs = ["checkpoint_ops_test.py"], 144 additional_deps = [ 145 "//third_party/py/numpy", 146 "//tensorflow/python:array_ops", 147 "//tensorflow/python:checkpoint_ops_gen", 148 "//tensorflow/python:client_testlib", 149 "//tensorflow/python:constant_op", 150 "//tensorflow/python:dtypes", 151 "//tensorflow/python:errors", 152 "//tensorflow/python:framework_ops", 153 "//tensorflow/python:math_ops", 154 "//tensorflow/python:partitioned_variables", 155 "//tensorflow/python:platform", 156 "//tensorflow/python:training", 157 "//tensorflow/python:variable_scope", 158 "//tensorflow/python:variables", 159 ], 160) 161 162cuda_py_test( 163 name = "cholesky_op_test", 164 size = "medium", 165 srcs = ["cholesky_op_test.py"], 166 additional_deps = [ 167 "//third_party/py/numpy", 168 "//tensorflow/python:array_ops", 169 "//tensorflow/python:client_testlib", 170 "//tensorflow/python:framework_for_generated_wrappers", 171 "//tensorflow/python:linalg_ops", 172 "//tensorflow/python:math_ops", 173 "//tensorflow/python:platform", 174 "//tensorflow/python/ops/linalg", 175 ], 176 shard_count = 5, 177 tags = ["no_windows_gpu"], 178) 179 180tf_py_test( 181 name = "clip_ops_test", 182 size = "small", 183 srcs = ["clip_ops_test.py"], 184 additional_deps = [ 185 "//tensorflow/python:client_testlib", 186 "//tensorflow/python:clip_ops", 187 "//tensorflow/python:framework_for_generated_wrappers", 188 ], 189 tags = ["no_windows"], 190) 191 192tf_py_test( 193 name = "conditional_accumulator_test", 194 size = "small", 195 srcs = ["conditional_accumulator_test.py"], 196 additional_deps = [ 197 "//third_party/py/numpy", 198 "//tensorflow/python:array_ops", 199 "//tensorflow/python:client_testlib", 200 "//tensorflow/python:data_flow_ops", 201 "//tensorflow/python:errors", 202 "//tensorflow/python:framework_for_generated_wrappers", 203 "//tensorflow/python:math_ops", 204 "//tensorflow/python:state_ops", 205 "//tensorflow/python:variables", 206 ], 207) 208 209tf_py_test( 210 name = "ctc_decoder_ops_test", 211 size = "small", 212 srcs = ["ctc_decoder_ops_test.py"], 213 additional_deps = [ 214 "//third_party/py/numpy", 215 "//tensorflow/python:array_ops", 216 "//tensorflow/python:client_testlib", 217 "//tensorflow/python:ctc_ops", 218 "//tensorflow/python:framework_for_generated_wrappers", 219 ], 220) 221 222tf_py_test( 223 name = "ctc_loss_op_test", 224 size = "small", 225 srcs = ["ctc_loss_op_test.py"], 226 additional_deps = [ 227 "//third_party/py/numpy", 228 "//tensorflow/python:client_testlib", 229 "//tensorflow/python:ctc_ops", 230 "//tensorflow/python:framework", 231 "//tensorflow/python:framework_for_generated_wrappers", 232 "//tensorflow/python:gradients", 233 ], 234) 235 236tf_py_test( 237 name = "decode_csv_op_test", 238 size = "small", 239 srcs = ["decode_csv_op_test.py"], 240 additional_deps = [ 241 "//third_party/py/numpy", 242 "//tensorflow/python:client_testlib", 243 "//tensorflow/python:parsing_ops", 244 ], 245) 246 247tf_py_test( 248 name = "decode_png_op_test", 249 size = "small", 250 srcs = ["decode_png_op_test.py"], 251 additional_deps = [ 252 "//tensorflow/python:array_ops", 253 "//tensorflow/python:client_testlib", 254 "//tensorflow/python:framework_for_generated_wrappers", 255 "//tensorflow/python:image_ops", 256 "//tensorflow/python:nn_grad", 257 ], 258) 259 260tf_py_test( 261 name = "decode_bmp_op_test", 262 size = "small", 263 srcs = ["decode_bmp_op_test.py"], 264 additional_deps = [ 265 "//tensorflow/python:array_ops", 266 "//tensorflow/python:client_testlib", 267 "//tensorflow/python:framework_for_generated_wrappers", 268 "//tensorflow/python:image_ops", 269 "//tensorflow/python:nn_grad", 270 ], 271) 272 273tf_py_test( 274 name = "decode_jpeg_op_test", 275 srcs = ["decode_jpeg_op_test.py"], 276 additional_deps = [ 277 "//tensorflow/python:client_testlib", 278 "//tensorflow/python:framework_for_generated_wrappers", 279 "//tensorflow/python:image_ops", 280 "//tensorflow/python:platform", 281 ], 282 data = ["//tensorflow/core:image_testdata"], 283) 284 285tf_py_test( 286 name = "decode_image_op_test", 287 size = "small", 288 srcs = ["decode_image_op_test.py"], 289 additional_deps = [ 290 "//third_party/py/numpy", 291 "//tensorflow/python:client_testlib", 292 "//tensorflow/python:errors", 293 "//tensorflow/python:image_ops", 294 "//tensorflow/python:io_ops", 295 "//tensorflow/python:nn_grad", 296 ], 297 data = ["//tensorflow/core:image_testdata"], 298 tags = ["no_windows"], 299) 300 301tf_py_test( 302 name = "decode_raw_op_test", 303 size = "small", 304 srcs = ["decode_raw_op_test.py"], 305 additional_deps = [ 306 "//third_party/py/numpy", 307 "//tensorflow/python:array_ops", 308 "//tensorflow/python:client_testlib", 309 "//tensorflow/python:framework_for_generated_wrappers", 310 "//tensorflow/python:parsing_ops", 311 ], 312) 313 314tf_py_test( 315 name = "decode_compressed_op_test", 316 size = "small", 317 srcs = ["decode_compressed_op_test.py"], 318 additional_deps = [ 319 "//third_party/py/numpy", 320 "//tensorflow/python:array_ops", 321 "//tensorflow/python:client_testlib", 322 "//tensorflow/python:framework_for_generated_wrappers", 323 "//tensorflow/python:parsing_ops", 324 ], 325) 326 327cuda_py_test( 328 name = "determinant_op_test", 329 size = "small", 330 srcs = ["determinant_op_test.py"], 331 additional_deps = [ 332 "//third_party/py/numpy", 333 "//tensorflow/python:client_testlib", 334 "//tensorflow/python:framework_for_generated_wrappers", 335 "//tensorflow/python:linalg_ops", 336 ], 337) 338 339tf_py_test( 340 name = "draw_bounding_box_op_test", 341 size = "small", 342 srcs = ["draw_bounding_box_op_test.py"], 343 additional_deps = [ 344 "//third_party/py/numpy", 345 "//tensorflow/python:array_ops", 346 "//tensorflow/python:client_testlib", 347 "//tensorflow/python:framework_for_generated_wrappers", 348 "//tensorflow/python:image_ops", 349 "//tensorflow/python:math_ops", 350 ], 351) 352 353tf_py_test( 354 name = "edit_distance_op_test", 355 size = "small", 356 srcs = ["edit_distance_op_test.py"], 357 additional_deps = [ 358 "//third_party/py/numpy", 359 "//tensorflow/python:array_ops", 360 "//tensorflow/python:client_testlib", 361 "//tensorflow/python:framework", 362 "//tensorflow/python:framework_for_generated_wrappers", 363 ], 364) 365 366tf_py_test( 367 name = "fifo_queue_test", 368 size = "small", 369 srcs = ["fifo_queue_test.py"], 370 additional_deps = [ 371 "//third_party/py/numpy", 372 "//tensorflow/core:protos_all_py", 373 "//tensorflow/python:array_ops", 374 "//tensorflow/python:client", 375 "//tensorflow/python:client_testlib", 376 "//tensorflow/python:data_flow_ops", 377 "//tensorflow/python:errors", 378 "//tensorflow/python:framework_for_generated_wrappers", 379 "//tensorflow/python:util", 380 ], 381) 382 383tf_py_test( 384 name = "fractional_avg_pool_op_test", 385 size = "small", 386 srcs = ["fractional_avg_pool_op_test.py"], 387 additional_deps = [ 388 "//third_party/py/numpy", 389 "//tensorflow/python:array_ops", 390 "//tensorflow/python:client_testlib", 391 "//tensorflow/python:framework_for_generated_wrappers", 392 "//tensorflow/python:nn_grad", 393 "//tensorflow/python:nn_ops", 394 "//tensorflow/python:nn_ops_gen", 395 ], 396) 397 398tf_py_test( 399 name = "fractional_max_pool_op_test", 400 size = "small", 401 srcs = ["fractional_max_pool_op_test.py"], 402 additional_deps = [ 403 "//third_party/py/numpy", 404 "//tensorflow/python:array_ops", 405 "//tensorflow/python:client_testlib", 406 "//tensorflow/python:framework_for_generated_wrappers", 407 "//tensorflow/python:nn_grad", 408 "//tensorflow/python:nn_ops", 409 "//tensorflow/python:nn_ops_gen", 410 ], 411) 412 413tf_py_test( 414 name = "identity_op_py_test", 415 size = "small", 416 srcs = ["identity_op_py_test.py"], 417 additional_deps = [ 418 "//third_party/py/numpy", 419 "//tensorflow/python:array_ops", 420 "//tensorflow/python:array_ops_gen", 421 "//tensorflow/python:client_testlib", 422 "//tensorflow/python:framework_for_generated_wrappers", 423 "//tensorflow/python:variables", 424 ], 425) 426 427tf_py_test( 428 name = "identity_n_op_py_test", 429 size = "small", 430 srcs = ["identity_n_op_py_test.py"], 431 additional_deps = [ 432 "//third_party/py/numpy", 433 "//tensorflow/python:array_ops", 434 "//tensorflow/python:array_ops_gen", 435 "//tensorflow/python:client_testlib", 436 "//tensorflow/python:framework_for_generated_wrappers", 437 "//tensorflow/python:variables", 438 ], 439) 440 441tf_py_test( 442 name = "in_topk_op_test", 443 size = "small", 444 srcs = ["in_topk_op_test.py"], 445 additional_deps = [ 446 "//third_party/py/numpy", 447 "//tensorflow/python:client_testlib", 448 "//tensorflow/python:errors", 449 "//tensorflow/python:nn_ops", 450 ], 451) 452 453tf_py_test( 454 name = "record_input_test", 455 size = "medium", 456 srcs = ["record_input_test.py"], 457 additional_deps = [ 458 "//tensorflow/python:client_testlib", 459 "//tensorflow/python:data_flow_ops", 460 "//tensorflow/python:io_ops", 461 "//tensorflow/python:util", 462 ], 463) 464 465tf_py_test( 466 name = "io_ops_test", 467 size = "small", 468 srcs = ["io_ops_test.py"], 469 additional_deps = [ 470 "//tensorflow/python:client_testlib", 471 "//tensorflow/python:io_ops", 472 "//tensorflow/python:util", 473 ], 474) 475 476tf_py_test( 477 name = "listdiff_op_test", 478 size = "small", 479 srcs = ["listdiff_op_test.py"], 480 additional_deps = [ 481 "//third_party/py/numpy", 482 "//tensorflow/python:array_ops", 483 "//tensorflow/python:client_testlib", 484 "//tensorflow/python:framework_for_generated_wrappers", 485 "//tensorflow/python:util", 486 ], 487) 488 489tf_py_test( 490 name = "logging_ops_test", 491 size = "small", 492 srcs = ["logging_ops_test.py"], 493 additional_deps = [ 494 "//tensorflow/python:client_testlib", 495 "//tensorflow/python:control_flow_ops", 496 "//tensorflow/python:framework_for_generated_wrappers", 497 "//tensorflow/python:framework_test_lib", 498 "//tensorflow/python:gradients", 499 "//tensorflow/python:logging_ops", 500 "//tensorflow/python:math_ops", 501 ], 502) 503 504tf_py_test( 505 name = "lookup_ops_test", 506 size = "small", 507 srcs = ["lookup_ops_test.py"], 508 additional_deps = [ 509 "//tensorflow/python:client", 510 "//tensorflow/python:client_testlib", 511 "//tensorflow/python:errors", 512 "//tensorflow/python:framework_for_generated_wrappers", 513 "//tensorflow/python:framework_test_lib", 514 "//tensorflow/python:lookup_ops", 515 "//tensorflow/python:sparse_tensor", 516 "//tensorflow/python:training", 517 ], 518 grpc_enabled = True, 519) 520 521tf_py_test( 522 name = "losses_test", 523 size = "medium", 524 srcs = ["losses_test.py"], 525 additional_deps = [ 526 "//third_party/py/numpy", 527 "//tensorflow/python/ops/losses", 528 "//tensorflow/python:array_ops", 529 "//tensorflow/python:client_testlib", 530 "//tensorflow/python:errors", 531 "//tensorflow/python:framework", 532 "//tensorflow/python:framework_for_generated_wrappers", 533 "//tensorflow/python:init_ops", 534 "//tensorflow/python:math_ops", 535 "//tensorflow/python:training", 536 "//tensorflow/python:variable_scope", 537 "//tensorflow/python:variables", 538 ], 539) 540 541tf_py_test( 542 name = "matrix_exponential_op_test", 543 size = "small", 544 srcs = ["matrix_exponential_op_test.py"], 545 additional_deps = [ 546 "//third_party/py/numpy", 547 "//tensorflow/python:client_testlib", 548 "//tensorflow/python:framework_for_generated_wrappers", 549 "//tensorflow/python:linalg_ops", 550 ], 551) 552 553tf_py_test( 554 name = "matrix_logarithm_op_test", 555 size = "small", 556 srcs = ["matrix_logarithm_op_test.py"], 557 additional_deps = [ 558 "//third_party/py/numpy", 559 "//tensorflow/python:client_testlib", 560 "//tensorflow/python:framework_for_generated_wrappers", 561 "//tensorflow/python:linalg_ops", 562 ], 563) 564 565cuda_py_test( 566 name = "matrix_inverse_op_test", 567 size = "small", 568 srcs = ["matrix_inverse_op_test.py"], 569 additional_deps = [ 570 "//third_party/py/numpy", 571 "//tensorflow/python:client_testlib", 572 "//tensorflow/python:framework_for_generated_wrappers", 573 "//tensorflow/python:linalg_ops", 574 "//tensorflow/python:math_ops", 575 ], 576) 577 578cuda_py_test( 579 name = "matrix_solve_ls_op_test", 580 size = "medium", 581 srcs = ["matrix_solve_ls_op_test.py"], 582 additional_deps = [ 583 "//third_party/py/numpy", 584 "//tensorflow/python:array_ops", 585 "//tensorflow/python:client_testlib", 586 "//tensorflow/python:framework_for_generated_wrappers", 587 "//tensorflow/python:linalg_ops", 588 "//tensorflow/python:math_ops", 589 ], 590) 591 592cuda_py_test( 593 name = "matrix_solve_op_test", 594 size = "small", 595 srcs = ["matrix_solve_op_test.py"], 596 additional_deps = [ 597 "//third_party/py/numpy", 598 "//tensorflow/python:array_ops", 599 "//tensorflow/python:client_testlib", 600 "//tensorflow/python:framework_for_generated_wrappers", 601 "//tensorflow/python:linalg_ops", 602 ], 603) 604 605cuda_py_test( 606 name = "matrix_triangular_solve_op_test", 607 size = "small", 608 srcs = ["matrix_triangular_solve_op_test.py"], 609 additional_deps = [ 610 "//third_party/py/numpy", 611 "//tensorflow/python:client_testlib", 612 "//tensorflow/python:linalg_ops", 613 ], 614) 615 616cuda_py_test( 617 name = "parameterized_truncated_normal_op_test", 618 size = "small", 619 srcs = ["parameterized_truncated_normal_op_test.py"], 620 additional_deps = [ 621 "//third_party/py/numpy", 622 "//tensorflow/core:protos_all_py", 623 "//tensorflow/python:client", 624 "//tensorflow/python:client_testlib", 625 "//tensorflow/python:control_flow_ops", 626 "//tensorflow/python:framework", 627 "//tensorflow/python:framework_for_generated_wrappers", 628 "//tensorflow/python:platform", 629 "//tensorflow/python:random_ops", 630 ], 631) 632 633tf_py_test( 634 name = "parsing_ops_test", 635 size = "small", 636 srcs = ["parsing_ops_test.py"], 637 additional_deps = [ 638 "//third_party/py/numpy", 639 "//tensorflow/core:protos_all_py", 640 "//tensorflow/python:array_ops", 641 "//tensorflow/python:client_testlib", 642 "//tensorflow/python:errors", 643 "//tensorflow/python:framework", 644 "//tensorflow/python:framework_for_generated_wrappers", 645 "//tensorflow/python:parsing_ops", 646 "//tensorflow/python:platform", 647 ], 648) 649 650tf_py_test( 651 name = "parse_single_example_op_test", 652 size = "small", 653 srcs = ["parse_single_example_op_test.py"], 654 additional_deps = [ 655 "//third_party/py/numpy", 656 "//tensorflow/core:protos_all_py", 657 "//tensorflow/python:array_ops", 658 "//tensorflow/python:client_testlib", 659 "//tensorflow/python:errors", 660 "//tensorflow/python:framework", 661 "//tensorflow/python:framework_for_generated_wrappers", 662 "//tensorflow/python:parsing_ops", 663 "//tensorflow/python:platform", 664 ], 665) 666 667tf_py_test( 668 name = "partitioned_variables_test", 669 size = "small", 670 srcs = ["partitioned_variables_test.py"], 671 additional_deps = [ 672 "//third_party/py/numpy", 673 "//tensorflow/python:array_ops", 674 "//tensorflow/python:client_testlib", 675 "//tensorflow/python:framework_for_generated_wrappers", 676 "//tensorflow/python:init_ops", 677 "//tensorflow/python:partitioned_variables", 678 "//tensorflow/python:random_ops", 679 "//tensorflow/python:variable_scope", 680 "//tensorflow/python:variables", 681 ], 682) 683 684tf_py_test( 685 name = "priority_queue_test", 686 size = "small", 687 srcs = ["priority_queue_test.py"], 688 additional_deps = [ 689 "//third_party/py/numpy", 690 "//tensorflow/python:array_ops", 691 "//tensorflow/python:client_testlib", 692 "//tensorflow/python:data_flow_ops", 693 "//tensorflow/python:errors", 694 "//tensorflow/python:framework_for_generated_wrappers", 695 "//tensorflow/python:nn_grad", 696 ], 697) 698 699cuda_py_test( 700 name = "resource_variable_ops_test", 701 size = "small", 702 srcs = ["resource_variable_ops_test.py"], 703 additional_deps = [ 704 "//tensorflow/python:array_ops", 705 "//tensorflow/python:constant_op", 706 "//tensorflow/python:client_testlib", 707 "//tensorflow/python:errors", 708 "//tensorflow/python:framework_for_generated_wrappers", 709 "//tensorflow/python:framework_test_lib", 710 "//tensorflow/python:resource_variable_ops", 711 "//tensorflow/python:variables", 712 ], 713) 714 715tf_py_test( 716 name = "save_restore_ops_test", 717 size = "small", 718 srcs = ["save_restore_ops_test.py"], 719 additional_deps = [ 720 "//tensorflow/core:protos_all_py", 721 "//tensorflow/python:client", 722 "//tensorflow/python:client_testlib", 723 "//tensorflow/python:io_ops", 724 "//tensorflow/python:io_ops_gen", 725 ], 726) 727 728cuda_py_test( 729 name = "scatter_nd_ops_test", 730 size = "medium", 731 srcs = ["scatter_nd_ops_test.py"], 732 additional_deps = [ 733 "//third_party/py/numpy", 734 "//tensorflow/python:array_ops", 735 "//tensorflow/python:client", 736 "//tensorflow/python:client_testlib", 737 "//tensorflow/python:framework_for_generated_wrappers", 738 "//tensorflow/python:gradients", 739 "//tensorflow/python:state_ops", 740 "//tensorflow/python:variables", 741 "//tensorflow/python:resource_variable_ops", 742 ], 743 tags = ["noasan"], # http://b/32635055 744) 745 746tf_py_test( 747 name = "segment_reduction_ops_test", 748 size = "medium", 749 srcs = ["segment_reduction_ops_test.py"], 750 additional_deps = [ 751 "//third_party/py/numpy", 752 "//tensorflow/python:client", 753 "//tensorflow/python:client_testlib", 754 "//tensorflow/python:framework_for_generated_wrappers", 755 "//tensorflow/python:math_ops", 756 "//tensorflow/python:variables", 757 "//tensorflow/python:nn_grad", 758 ], 759) 760 761tf_py_test( 762 name = "sparse_add_op_test", 763 size = "small", 764 srcs = ["sparse_add_op_test.py"], 765 additional_deps = [ 766 "//third_party/py/numpy", 767 "//tensorflow/python:client", 768 "//tensorflow/python:client_testlib", 769 "//tensorflow/python:framework", 770 "//tensorflow/python:framework_for_generated_wrappers", 771 "//tensorflow/python:math_ops", 772 "//tensorflow/python:sparse_grad", 773 "//tensorflow/python:sparse_ops", 774 ], 775) 776 777tf_py_test( 778 name = "sparse_concat_op_test", 779 size = "small", 780 srcs = ["sparse_concat_op_test.py"], 781 additional_deps = [ 782 "//third_party/py/numpy", 783 "//tensorflow/python:array_ops", 784 "//tensorflow/python:client_testlib", 785 "//tensorflow/python:framework", 786 "//tensorflow/python:framework_for_generated_wrappers", 787 "//tensorflow/python:sparse_ops", 788 ], 789) 790 791tf_py_test( 792 name = "sparse_conditional_accumulator_test", 793 size = "small", 794 srcs = ["sparse_conditional_accumulator_test.py"], 795 additional_deps = [ 796 "//third_party/py/numpy", 797 "//tensorflow/python:array_ops", 798 "//tensorflow/python:client_testlib", 799 "//tensorflow/python:data_flow_ops", 800 "//tensorflow/python:errors", 801 "//tensorflow/python:framework_for_generated_wrappers", 802 ], 803) 804 805tf_py_test( 806 name = "sparse_reorder_op_test", 807 size = "small", 808 srcs = ["sparse_reorder_op_test.py"], 809 additional_deps = [ 810 "//third_party/py/numpy", 811 "//tensorflow/python:array_ops", 812 "//tensorflow/python:client_testlib", 813 "//tensorflow/python:framework", 814 "//tensorflow/python:framework_for_generated_wrappers", 815 "//tensorflow/python:sparse_grad", 816 "//tensorflow/python:sparse_ops", 817 ], 818) 819 820tf_py_test( 821 name = "sparse_reshape_op_test", 822 size = "small", 823 srcs = ["sparse_reshape_op_test.py"], 824 additional_deps = [ 825 "//third_party/py/numpy", 826 "//tensorflow/python:array_ops", 827 "//tensorflow/python:client_testlib", 828 "//tensorflow/python:framework", 829 "//tensorflow/python:framework_for_generated_wrappers", 830 "//tensorflow/python:sparse_ops", 831 ], 832) 833 834tf_py_test( 835 name = "sparse_split_op_test", 836 size = "small", 837 srcs = ["sparse_split_op_test.py"], 838 additional_deps = [ 839 "//third_party/py/numpy", 840 "//tensorflow/python:client_testlib", 841 "//tensorflow/python:framework", 842 "//tensorflow/python:sparse_ops", 843 ], 844) 845 846tf_py_test( 847 name = "sparse_slice_op_test", 848 size = "small", 849 srcs = ["sparse_slice_op_test.py"], 850 additional_deps = [ 851 "//third_party/py/numpy", 852 "//tensorflow/python:client_testlib", 853 "//tensorflow/python:framework", 854 "//tensorflow/python:sparse_ops", 855 ], 856) 857 858tf_py_test( 859 name = "sparse_to_dense_op_py_test", 860 size = "small", 861 srcs = ["sparse_to_dense_op_py_test.py"], 862 additional_deps = [ 863 "//third_party/py/numpy", 864 "//tensorflow/python:array_ops", 865 "//tensorflow/python:client_testlib", 866 "//tensorflow/python:framework_for_generated_wrappers", 867 "//tensorflow/python:sparse_ops", 868 ], 869) 870 871tf_py_test( 872 name = "sparsemask_op_test", 873 size = "small", 874 srcs = ["sparsemask_op_test.py"], 875 additional_deps = [ 876 "//third_party/py/numpy", 877 "//tensorflow/python:array_ops", 878 "//tensorflow/python:client_testlib", 879 "//tensorflow/python:framework_for_generated_wrappers", 880 ], 881) 882 883tf_py_test( 884 name = "string_join_op_test", 885 size = "small", 886 srcs = ["string_join_op_test.py"], 887 additional_deps = [ 888 "//tensorflow/python:client_testlib", 889 "//tensorflow/python:string_ops", 890 ], 891) 892 893tf_py_test( 894 name = "string_split_op_test", 895 size = "small", 896 srcs = ["string_split_op_test.py"], 897 additional_deps = [ 898 "//third_party/py/numpy", 899 "//tensorflow/python:array_ops", 900 "//tensorflow/python:client_testlib", 901 "//tensorflow/python:errors", 902 "//tensorflow/python:framework_for_generated_wrappers", 903 "//tensorflow/python:string_ops", 904 ], 905) 906 907tf_py_test( 908 name = "substr_op_test", 909 size = "small", 910 srcs = ["substr_op_test.py"], 911 additional_deps = [ 912 "//third_party/py/numpy", 913 "//tensorflow/python:client_testlib", 914 "//tensorflow/python:errors", 915 "//tensorflow/python:string_ops", 916 ], 917) 918 919tf_py_test( 920 name = "summary_ops_test", 921 size = "small", 922 srcs = ["summary_ops_test.py"], 923 additional_deps = [ 924 "//tensorflow/core:protos_all_py", 925 "//tensorflow/python:client_testlib", 926 "//tensorflow/python:framework_for_generated_wrappers", 927 "//tensorflow/python:logging_ops", 928 "//tensorflow/python:summary", 929 ], 930) 931 932tf_py_test( 933 name = "summary_tensor_op_test", 934 size = "small", 935 srcs = ["summary_tensor_op_test.py"], 936 additional_deps = [ 937 "//third_party/py/numpy", 938 "@six_archive//:six", 939 "//tensorflow/core:protos_all_py", 940 "//tensorflow/python:array_ops", 941 "//tensorflow/python:client_testlib", 942 "//tensorflow/python:framework", 943 "//tensorflow/python:framework_for_generated_wrappers", 944 "//tensorflow/python:summary_ops", 945 ], 946) 947 948tf_py_test( 949 name = "template_test", 950 size = "small", 951 srcs = ["template_test.py"], 952 additional_deps = [ 953 "//tensorflow/python:client", 954 "//tensorflow/python:client_testlib", 955 "//tensorflow/python:framework", 956 "//tensorflow/python:init_ops", 957 "//tensorflow/python:math_ops", 958 "//tensorflow/python:nn_grad", 959 "//tensorflow/python:template", 960 "//tensorflow/python:training", 961 "//tensorflow/python:variable_scope", 962 "//tensorflow/python:variables", 963 ], 964) 965 966cuda_py_test( 967 name = "topk_op_test", 968 size = "small", 969 srcs = ["topk_op_test.py"], 970 additional_deps = [ 971 "//third_party/py/numpy", 972 "//tensorflow/python:array_ops", 973 "//tensorflow/python:client_testlib", 974 "//tensorflow/python:framework_for_generated_wrappers", 975 "//tensorflow/python:gradients", 976 "//tensorflow/python:nn_grad", 977 "//tensorflow/python:nn_ops", 978 ], 979) 980 981cuda_py_test( 982 name = "nth_element_op_test", 983 size = "small", 984 srcs = ["nth_element_op_test.py"], 985 additional_deps = [ 986 "//third_party/py/numpy", 987 "//tensorflow/python:array_ops", 988 "//tensorflow/python:client_testlib", 989 "//tensorflow/python:framework_for_generated_wrappers", 990 "//tensorflow/python:gradients", 991 "//tensorflow/python:nn_grad", 992 "//tensorflow/python:nn_ops", 993 ], 994) 995 996tf_py_test( 997 name = "unique_op_test", 998 size = "small", 999 srcs = ["unique_op_test.py"], 1000 additional_deps = [ 1001 "//third_party/py/numpy", 1002 "//tensorflow/python:array_ops", 1003 "//tensorflow/python:client_testlib", 1004 ], 1005) 1006 1007tf_py_test( 1008 name = "variable_scope_test", 1009 size = "small", 1010 srcs = ["variable_scope_test.py"], 1011 additional_deps = [ 1012 "//third_party/py/numpy", 1013 "//tensorflow/python:client_testlib", 1014 "//tensorflow/python:control_flow_ops", 1015 "//tensorflow/python:errors", 1016 "//tensorflow/python:framework_for_generated_wrappers", 1017 "//tensorflow/python:framework_test_lib", 1018 "//tensorflow/python:init_ops", 1019 "//tensorflow/python:math_ops", 1020 "//tensorflow/python:variable_scope", 1021 "//tensorflow/python:resource_variable_ops", 1022 "//tensorflow/python:state_ops", 1023 "//tensorflow/python:variables", 1024 "//tensorflow/python/eager:context", 1025 ], 1026 tags = ["no_windows"], 1027) 1028 1029tf_py_test( 1030 name = "variables_test", 1031 size = "small", 1032 srcs = ["variables_test.py"], 1033 additional_deps = [ 1034 "//third_party/py/numpy", 1035 "//tensorflow/python:array_ops", 1036 "//tensorflow/python:client_testlib", 1037 "//tensorflow/python:control_flow_ops", 1038 "//tensorflow/python:errors", 1039 "//tensorflow/python:framework_for_generated_wrappers", 1040 "//tensorflow/python:math_ops", 1041 "//tensorflow/python:random_ops", 1042 "//tensorflow/python:state_ops_gen", 1043 "//tensorflow/python:training", 1044 "//tensorflow/python:util", 1045 "//tensorflow/python:variables", 1046 "//tensorflow/python/eager:function", 1047 ], 1048) 1049 1050cuda_py_test( 1051 name = "where_op_test", 1052 size = "medium", 1053 srcs = ["where_op_test.py"], 1054 additional_deps = [ 1055 "//third_party/py/numpy", 1056 "//tensorflow/python:array_ops", 1057 "//tensorflow/python:client_testlib", 1058 "//tensorflow/python:framework_for_generated_wrappers", 1059 ], 1060) 1061 1062cuda_py_test( 1063 name = "cast_op_test", 1064 size = "small", 1065 srcs = ["cast_op_test.py"], 1066 additional_deps = [ 1067 "//third_party/py/numpy", 1068 "//tensorflow/python:array_ops", 1069 "//tensorflow/python:client_testlib", 1070 "//tensorflow/python:framework", 1071 "//tensorflow/python:framework_for_generated_wrappers", 1072 "//tensorflow/python:math_ops", 1073 "//tensorflow/python:variables", 1074 ], 1075 tags = [ 1076 "no_windows", 1077 "noasan", 1078 ], 1079) 1080 1081cuda_py_test( 1082 name = "dense_update_ops_no_tsan_test", 1083 size = "small", 1084 srcs = ["dense_update_ops_no_tsan_test.py"], 1085 additional_deps = [ 1086 "//third_party/py/numpy", 1087 "//tensorflow/python:array_ops", 1088 "//tensorflow/python:client_testlib", 1089 "//tensorflow/python:math_ops", 1090 "//tensorflow/python:state_ops", 1091 "//tensorflow/python:variables", 1092 ], 1093 tags = ["notsan"], 1094) 1095 1096cuda_py_test( 1097 name = "diag_op_test", 1098 size = "medium", 1099 srcs = ["diag_op_test.py"], 1100 additional_deps = [ 1101 "//third_party/py/numpy", 1102 "//tensorflow/python:array_ops", 1103 "//tensorflow/python:client_testlib", 1104 "//tensorflow/python:framework_for_generated_wrappers", 1105 "//tensorflow/python:gradients", 1106 "//tensorflow/python:platform", 1107 ], 1108 shard_count = 2, 1109 tags = ["no_windows_gpu"], 1110) 1111 1112tf_py_test( 1113 name = "reader_ops_test", 1114 size = "small", 1115 srcs = ["reader_ops_test.py"], 1116 additional_deps = [ 1117 "@six_archive//:six", 1118 "//tensorflow/core:protos_all_py", 1119 "//tensorflow/python:client_testlib", 1120 "//tensorflow/python:data_flow_ops", 1121 "//tensorflow/python:errors", 1122 "//tensorflow/python:framework_for_generated_wrappers", 1123 "//tensorflow/python:io_ops", 1124 "//tensorflow/python:lib", 1125 "//tensorflow/python:util", 1126 "//tensorflow/python:variables", 1127 ], 1128 data = ["//tensorflow/core:lmdb_testdata"], 1129 tags = ["no_windows"], 1130) 1131 1132cuda_py_test( 1133 name = "aggregate_ops_test", 1134 size = "small", 1135 srcs = ["aggregate_ops_test.py"], 1136 additional_deps = [ 1137 "//third_party/py/numpy", 1138 "//tensorflow/python:array_ops", 1139 "//tensorflow/python:client_testlib", 1140 "//tensorflow/python:framework_for_generated_wrappers", 1141 "//tensorflow/python:math_ops", 1142 ], 1143) 1144 1145cuda_py_test( 1146 name = "argmax_op_test", 1147 size = "small", 1148 srcs = ["argmax_op_test.py"], 1149 additional_deps = [ 1150 "//third_party/py/numpy", 1151 "//tensorflow/python:client_testlib", 1152 "//tensorflow/python:math_ops", 1153 ], 1154) 1155 1156cuda_py_test( 1157 name = "array_ops_test", 1158 size = "medium", 1159 srcs = ["array_ops_test.py"], 1160 additional_deps = [ 1161 "//third_party/py/numpy", 1162 "//tensorflow/python:array_ops", 1163 "//tensorflow/python:client", 1164 "//tensorflow/python:client_testlib", 1165 "//tensorflow/python:errors", 1166 "//tensorflow/python:framework", 1167 "//tensorflow/python:framework_for_generated_wrappers", 1168 "//tensorflow/python:framework_test_lib", 1169 "//tensorflow/python:gradients", 1170 "//tensorflow/python:math_ops", 1171 "//tensorflow/python:state_ops", 1172 "//tensorflow/python:test_ops", 1173 "//tensorflow/python:variables", 1174 "//tensorflow/python/eager:context", 1175 ], 1176 shard_count = 10, 1177) 1178 1179cuda_py_test( 1180 name = "batch_matmul_op_test", 1181 size = "small", 1182 srcs = ["batch_matmul_op_test.py"], 1183 additional_deps = [ 1184 "//third_party/py/numpy", 1185 "//tensorflow/python:array_ops", 1186 "//tensorflow/python:client_testlib", 1187 "//tensorflow/python:framework_for_generated_wrappers", 1188 "//tensorflow/python:math_ops", 1189 ], 1190 shard_count = 20, 1191) 1192 1193cuda_py_test( 1194 name = "batchtospace_op_test", 1195 size = "small", 1196 srcs = ["batchtospace_op_test.py"], 1197 additional_deps = [ 1198 "//third_party/py/numpy", 1199 "//tensorflow/python:array_ops", 1200 "//tensorflow/python:array_ops_gen", 1201 "//tensorflow/python:client_testlib", 1202 "//tensorflow/python:framework_for_generated_wrappers", 1203 ], 1204) 1205 1206cuda_py_test( 1207 name = "betainc_op_test", 1208 size = "small", 1209 srcs = ["betainc_op_test.py"], 1210 additional_deps = [ 1211 "//third_party/py/numpy", 1212 "//tensorflow/python:array_ops", 1213 "//tensorflow/python:client_testlib", 1214 "//tensorflow/python:framework_for_generated_wrappers", 1215 "//tensorflow/python:math_ops", 1216 "//tensorflow/python:platform", 1217 ], 1218) 1219 1220cuda_py_test( 1221 name = "bias_op_test", 1222 size = "small", 1223 srcs = ["bias_op_test.py"], 1224 additional_deps = [ 1225 "//third_party/py/numpy", 1226 "//tensorflow/python:array_ops", 1227 "//tensorflow/python:client_testlib", 1228 "//tensorflow/python:framework_for_generated_wrappers", 1229 "//tensorflow/python:gradients", 1230 "//tensorflow/python:nn_grad", 1231 "//tensorflow/python:nn_ops", 1232 ], 1233) 1234 1235cuda_py_test( 1236 name = "bitcast_op_test", 1237 size = "small", 1238 srcs = ["bitcast_op_test.py"], 1239 additional_deps = [ 1240 "//third_party/py/numpy", 1241 "//tensorflow/python:array_ops", 1242 "//tensorflow/python:client_testlib", 1243 "//tensorflow/python:framework_for_generated_wrappers", 1244 ], 1245) 1246 1247cuda_py_test( 1248 name = "check_ops_test", 1249 size = "small", 1250 srcs = ["check_ops_test.py"], 1251 additional_deps = [ 1252 "//third_party/py/numpy", 1253 "//tensorflow/python/eager:context", 1254 "//tensorflow/python:array_ops", 1255 "//tensorflow/python:check_ops", 1256 "//tensorflow/python:client_testlib", 1257 "//tensorflow/python:framework", 1258 "//tensorflow/python:framework_for_generated_wrappers", 1259 ], 1260) 1261 1262cuda_py_test( 1263 name = "constant_op_test", 1264 size = "small", 1265 srcs = ["constant_op_test.py"], 1266 additional_deps = [ 1267 "//third_party/py/numpy", 1268 "//tensorflow/python:array_ops", 1269 "//tensorflow/python:client_testlib", 1270 "//tensorflow/python:errors", 1271 "//tensorflow/python:framework_for_generated_wrappers", 1272 "//tensorflow/python:math_ops", 1273 "//tensorflow/python:util", 1274 ], 1275) 1276 1277cuda_py_test( 1278 name = "constant_op_eager_test", 1279 size = "small", 1280 srcs = ["constant_op_eager_test.py"], 1281 additional_deps = [ 1282 "//tensorflow/python/eager:core", 1283 "//tensorflow/python/eager:context", 1284 "//tensorflow/python/eager:test", 1285 "//third_party/py/numpy", 1286 "//tensorflow/python:array_ops", 1287 "//tensorflow/python:client_testlib", 1288 "//tensorflow/python:errors", 1289 "//tensorflow/python:framework_for_generated_wrappers", 1290 "//tensorflow/python:math_ops", 1291 "//tensorflow/python:util", 1292 ], 1293) 1294 1295cuda_py_test( 1296 name = "control_flow_ops_py_test", 1297 # TODO(b/70473603): change this back to "small" once the C API is 1298 # permanently enabled 1299 size = "medium", 1300 srcs = ["control_flow_ops_py_test.py"], 1301 additional_deps = [ 1302 "//third_party/py/numpy", 1303 "//tensorflow/core:protos_all_py", 1304 "//tensorflow/python:array_ops", 1305 "//tensorflow/python:array_ops_gen", 1306 "//tensorflow/python:client", 1307 "//tensorflow/python:client_testlib", 1308 "//tensorflow/python:control_flow_ops", 1309 "//tensorflow/python:data_flow_ops", 1310 "//tensorflow/python:data_flow_ops_gen", 1311 "//tensorflow/python:distributed_framework_test_lib", 1312 "//tensorflow/python:errors", 1313 "//tensorflow/python:framework", 1314 "//tensorflow/python:framework_for_generated_wrappers", 1315 "//tensorflow/python:functional_ops", 1316 "//tensorflow/python:gradients", 1317 "//tensorflow/python:logging_ops", 1318 "//tensorflow/python:logging_ops_gen", 1319 "//tensorflow/python:math_ops", 1320 "//tensorflow/python:nn_grad", 1321 "//tensorflow/python:resource_variable_ops", 1322 "//tensorflow/python:script_ops", 1323 "//tensorflow/python:state_ops", 1324 "//tensorflow/python:state_ops_gen", 1325 "//tensorflow/python:tensor_array_grad", 1326 "//tensorflow/python:training", 1327 "//tensorflow/python:util", 1328 "//tensorflow/python:variable_scope", 1329 "//tensorflow/python:variables", 1330 ], 1331) 1332 1333tf_py_test( 1334 name = "control_flow_util_test", 1335 size = "small", 1336 srcs = ["control_flow_util_test.py"], 1337 additional_deps = [ 1338 "//tensorflow/python:client_testlib", 1339 "//tensorflow/python:control_flow_ops", 1340 "//tensorflow/python:control_flow_ops_gen", 1341 "//tensorflow/python:control_flow_util", 1342 "//tensorflow/python:test_ops", 1343 ], 1344) 1345 1346cuda_py_test( 1347 name = "conv1d_test", 1348 size = "small", 1349 srcs = ["conv1d_test.py"], 1350 additional_deps = [ 1351 "//tensorflow/python:array_ops", 1352 "//tensorflow/python:client_testlib", 1353 "//tensorflow/python:framework_for_generated_wrappers", 1354 "//tensorflow/python:nn_ops", 1355 ], 1356) 1357 1358cuda_py_test( 1359 name = "conv2d_transpose_test", 1360 size = "small", 1361 srcs = ["conv2d_transpose_test.py"], 1362 additional_deps = [ 1363 "//third_party/py/numpy", 1364 "//tensorflow/python:client", 1365 "//tensorflow/python:client_testlib", 1366 "//tensorflow/python:framework_for_generated_wrappers", 1367 "//tensorflow/python:nn_grad", 1368 "//tensorflow/python:nn_ops", 1369 ], 1370) 1371 1372cuda_py_test( 1373 name = "conv3d_backprop_filter_v2_grad_test", 1374 size = "small", 1375 srcs = ["conv3d_backprop_filter_v2_grad_test.py"], 1376 additional_deps = [ 1377 "//third_party/py/numpy", 1378 "//tensorflow/python:array_ops", 1379 "//tensorflow/python:client_testlib", 1380 "//tensorflow/python:framework_for_generated_wrappers", 1381 "//tensorflow/python:nn_grad", 1382 "//tensorflow/python:nn_ops", 1383 ], 1384) 1385 1386cuda_py_test( 1387 name = "cross_grad_test", 1388 size = "small", 1389 srcs = ["cross_grad_test.py"], 1390 additional_deps = [ 1391 "//tensorflow/python:array_ops", 1392 "//tensorflow/python:client_testlib", 1393 "//tensorflow/python:math_ops", 1394 ], 1395) 1396 1397cuda_py_test( 1398 name = "denormal_test", 1399 size = "small", 1400 srcs = ["denormal_test.py"], 1401 additional_deps = [ 1402 "//third_party/py/numpy", 1403 "//tensorflow/python:array_ops", 1404 "//tensorflow/python:client_testlib", 1405 "//tensorflow/python:framework_for_generated_wrappers", 1406 "//tensorflow/python:platform", 1407 ], 1408) 1409 1410cuda_py_test( 1411 name = "dense_update_ops_test", 1412 size = "small", 1413 srcs = ["dense_update_ops_test.py"], 1414 additional_deps = [ 1415 "//third_party/py/numpy", 1416 "//tensorflow/python:array_ops", 1417 "//tensorflow/python:client_testlib", 1418 "//tensorflow/python:framework_for_generated_wrappers", 1419 "//tensorflow/python:math_ops", 1420 "//tensorflow/python:state_ops", 1421 "//tensorflow/python:variables", 1422 ], 1423) 1424 1425cuda_py_test( 1426 name = "depthtospace_op_test", 1427 size = "medium", 1428 srcs = ["depthtospace_op_test.py"], 1429 additional_deps = [ 1430 "//third_party/py/numpy", 1431 "//tensorflow/python:array_ops", 1432 "//tensorflow/python:client_testlib", 1433 "//tensorflow/python:framework_for_generated_wrappers", 1434 "//tensorflow/python:math_ops", 1435 ], 1436) 1437 1438cuda_py_test( 1439 name = "division_past_test", 1440 size = "medium", 1441 srcs = ["division_past_test.py"], 1442 additional_deps = [ 1443 "//third_party/py/numpy", 1444 "//tensorflow/python:client_testlib", 1445 "//tensorflow/python:framework_for_generated_wrappers", 1446 ], 1447 tags = ["manual"], 1448) 1449 1450cuda_py_test( 1451 name = "dynamic_partition_op_test", 1452 size = "medium", 1453 srcs = ["dynamic_partition_op_test.py"], 1454 additional_deps = [ 1455 "//third_party/py/numpy", 1456 "//tensorflow/python:array_ops", 1457 "//tensorflow/python:client_testlib", 1458 "//tensorflow/python:data_flow_grad", 1459 "//tensorflow/python:data_flow_ops", 1460 "//tensorflow/python:framework_for_generated_wrappers", 1461 "//tensorflow/python:gradients", 1462 ], 1463) 1464 1465cuda_py_test( 1466 name = "dynamic_stitch_op_test", 1467 size = "small", 1468 srcs = ["dynamic_stitch_op_test.py"], 1469 additional_deps = [ 1470 "//third_party/py/numpy", 1471 "//tensorflow/python:client_testlib", 1472 "//tensorflow/python:data_flow_grad", 1473 "//tensorflow/python:data_flow_ops", 1474 "//tensorflow/python:framework_for_generated_wrappers", 1475 "//tensorflow/python:gradients", 1476 ], 1477) 1478 1479cuda_py_test( 1480 name = "extract_image_patches_op_test", 1481 size = "small", 1482 srcs = ["extract_image_patches_op_test.py"], 1483 additional_deps = [ 1484 "//third_party/py/numpy", 1485 "//tensorflow/python:array_ops", 1486 "//tensorflow/python:client_testlib", 1487 "//tensorflow/python:framework_for_generated_wrappers", 1488 ], 1489) 1490 1491cuda_py_test( 1492 name = "functional_ops_test", 1493 size = "small", 1494 srcs = ["functional_ops_test.py"], 1495 additional_deps = [ 1496 "//third_party/py/numpy", 1497 "//tensorflow/python:array_ops", 1498 "//tensorflow/python:client_testlib", 1499 "//tensorflow/python:framework", 1500 "//tensorflow/python:framework_for_generated_wrappers", 1501 "//tensorflow/python:functional_ops", 1502 "//tensorflow/python:gradients", 1503 "//tensorflow/python:init_ops", 1504 "//tensorflow/python:math_ops", 1505 "//tensorflow/python:tensor_array_grad", 1506 "//tensorflow/python:variable_scope", 1507 "//tensorflow/python:variables", 1508 ], 1509 grpc_enabled = True, 1510 tags = ["no_windows"], 1511) 1512 1513cuda_py_test( 1514 name = "gather_nd_op_test", 1515 size = "small", 1516 srcs = ["gather_nd_op_test.py"], 1517 additional_deps = [ 1518 "//third_party/py/numpy", 1519 "//tensorflow/python:array_ops", 1520 "//tensorflow/python:client", 1521 "//tensorflow/python:client_testlib", 1522 "//tensorflow/python:framework_for_generated_wrappers", 1523 "//tensorflow/python:gradients", 1524 "//tensorflow/python:variables", 1525 ], 1526) 1527 1528cuda_py_test( 1529 name = "gather_op_test", 1530 size = "medium", 1531 srcs = ["gather_op_test.py"], 1532 additional_deps = [ 1533 "//third_party/py/numpy", 1534 "//tensorflow/python:array_ops", 1535 "//tensorflow/python:client_testlib", 1536 "//tensorflow/python:framework_for_generated_wrappers", 1537 "//tensorflow/python:gradients", 1538 ], 1539) 1540 1541cuda_py_test( 1542 name = "gradient_correctness_test", 1543 size = "small", 1544 srcs = ["gradient_correctness_test.py"], 1545 additional_deps = [ 1546 "//third_party/py/numpy", 1547 "//tensorflow/python:client_testlib", 1548 "//tensorflow/python:framework_for_generated_wrappers", 1549 "//tensorflow/python:gradients", 1550 "//tensorflow/python:math_ops", 1551 ], 1552) 1553 1554cuda_py_test( 1555 name = "init_ops_test", 1556 size = "small", 1557 srcs = ["init_ops_test.py"], 1558 additional_deps = [ 1559 "//third_party/py/numpy", 1560 "//tensorflow/python:array_ops", 1561 "//tensorflow/python:client_testlib", 1562 "//tensorflow/python:framework", 1563 "//tensorflow/python:framework_for_generated_wrappers", 1564 "//tensorflow/python:init_ops", 1565 "//tensorflow/python:math_ops", 1566 "//tensorflow/python:nn_ops", 1567 "//tensorflow/python:partitioned_variables", 1568 "//tensorflow/python:variable_scope", 1569 "//tensorflow/python:variables", 1570 ], 1571) 1572 1573cuda_py_test( 1574 name = "linalg_ops_test", 1575 size = "small", 1576 srcs = ["linalg_ops_test.py"], 1577 additional_deps = [ 1578 "//third_party/py/numpy", 1579 "//tensorflow/python:array_ops", 1580 "//tensorflow/python:client_testlib", 1581 "//tensorflow/python:framework_for_generated_wrappers", 1582 "//tensorflow/python:linalg_ops", 1583 "//tensorflow/python:math_ops", 1584 "//tensorflow/python/ops/linalg", 1585 ], 1586 tags = ["no_windows_gpu"], 1587) 1588 1589cuda_py_test( 1590 name = "lrn_op_test", 1591 size = "small", 1592 srcs = ["lrn_op_test.py"], 1593 additional_deps = [ 1594 "//third_party/py/numpy", 1595 "//tensorflow/python:array_ops", 1596 "//tensorflow/python:client_testlib", 1597 "//tensorflow/python:framework_for_generated_wrappers", 1598 "//tensorflow/python:gradients", 1599 "//tensorflow/python:nn", 1600 "//tensorflow/python:nn_grad", 1601 ], 1602) 1603 1604cuda_py_test( 1605 name = "manip_ops_test", 1606 size = "small", 1607 srcs = ["manip_ops_test.py"], 1608 additional_deps = [ 1609 "//third_party/py/numpy", 1610 "//tensorflow/python:manip_ops", 1611 "//tensorflow/python:client_testlib", 1612 "//tensorflow/python:framework_for_generated_wrappers", 1613 ], 1614 tags = ["no_windows_gpu"], 1615) 1616 1617cuda_py_test( 1618 name = "matmul_op_test", 1619 size = "small", 1620 srcs = ["matmul_op_test.py"], 1621 additional_deps = [ 1622 "//third_party/py/numpy", 1623 "//tensorflow/python:array_ops", 1624 "//tensorflow/python:client_testlib", 1625 "//tensorflow/python:framework_for_generated_wrappers", 1626 "//tensorflow/python:framework_test_lib", 1627 "//tensorflow/python:math_ops", 1628 "//tensorflow/python:random_ops", 1629 "//tensorflow/python:variables", 1630 ], 1631 shard_count = 20, 1632) 1633 1634cuda_py_test( 1635 name = "morphological_ops_test", 1636 size = "small", 1637 srcs = ["morphological_ops_test.py"], 1638 additional_deps = [ 1639 "//third_party/py/numpy", 1640 "//tensorflow/python:client_testlib", 1641 "//tensorflow/python:framework_for_generated_wrappers", 1642 "//tensorflow/python:nn_grad", 1643 "//tensorflow/python:nn_ops", 1644 ], 1645) 1646 1647cuda_py_test( 1648 name = "numerics_test", 1649 size = "small", 1650 srcs = ["numerics_test.py"], 1651 additional_deps = [ 1652 "//third_party/py/numpy", 1653 "//tensorflow/python:array_ops", 1654 "//tensorflow/python:client_testlib", 1655 "//tensorflow/python:control_flow_ops", 1656 "//tensorflow/python:framework_for_generated_wrappers", 1657 "//tensorflow/python:math_ops", 1658 "//tensorflow/python:numerics", 1659 ], 1660) 1661 1662cuda_py_test( 1663 name = "one_hot_op_test", 1664 size = "small", 1665 srcs = ["one_hot_op_test.py"], 1666 additional_deps = [ 1667 "//third_party/py/numpy", 1668 "//tensorflow/python:array_ops", 1669 "//tensorflow/python:client_testlib", 1670 "//tensorflow/python:framework_for_generated_wrappers", 1671 ], 1672 tags = ["no_windows_gpu"], 1673) 1674 1675cuda_py_test( 1676 name = "stack_op_test", 1677 size = "small", 1678 srcs = ["stack_op_test.py"], 1679 additional_deps = [ 1680 "//third_party/py/numpy", 1681 "//tensorflow/python:array_ops", 1682 "//tensorflow/python:client_testlib", 1683 "//tensorflow/python:errors", 1684 "//tensorflow/python:framework_for_generated_wrappers", 1685 "//tensorflow/python:variables", 1686 ], 1687) 1688 1689cuda_py_test( 1690 name = "pad_op_test", 1691 size = "small", 1692 srcs = ["pad_op_test.py"], 1693 additional_deps = [ 1694 "//third_party/py/numpy", 1695 "//tensorflow/python:array_ops", 1696 "//tensorflow/python:client_testlib", 1697 "//tensorflow/python:framework_for_generated_wrappers", 1698 ], 1699) 1700 1701cuda_py_test( 1702 name = "padding_fifo_queue_test", 1703 size = "small", 1704 srcs = ["padding_fifo_queue_test.py"], 1705 additional_deps = [ 1706 "//third_party/py/numpy", 1707 "//tensorflow/python:array_ops", 1708 "//tensorflow/python:client_testlib", 1709 "//tensorflow/python:data_flow_ops", 1710 "//tensorflow/python:errors", 1711 "//tensorflow/python:framework_for_generated_wrappers", 1712 ], 1713) 1714 1715cuda_py_test( 1716 name = "py_func_test", 1717 size = "small", 1718 srcs = ["py_func_test.py"], 1719 additional_deps = [ 1720 "//third_party/py/numpy", 1721 "//tensorflow/python:array_ops", 1722 "//tensorflow/python:client", 1723 "//tensorflow/python:client_testlib", 1724 "//tensorflow/python:errors", 1725 "//tensorflow/python:framework_for_generated_wrappers", 1726 "//tensorflow/python:script_ops", 1727 "//tensorflow/python/eager:context", 1728 "//tensorflow/python/eager:function", 1729 ], 1730 tags = ["no_windows"], 1731) 1732 1733cuda_py_test( 1734 name = "reduce_join_op_test", 1735 size = "small", 1736 srcs = ["reduce_join_op_test.py"], 1737 additional_deps = [ 1738 "//third_party/py/numpy", 1739 "//tensorflow/python:array_ops", 1740 "//tensorflow/python:client_testlib", 1741 "//tensorflow/python:framework_for_generated_wrappers", 1742 "//tensorflow/python:string_ops", 1743 ], 1744) 1745 1746cuda_py_test( 1747 name = "reduction_ops_test", 1748 size = "medium", 1749 srcs = ["reduction_ops_test.py"], 1750 additional_deps = [ 1751 "//third_party/py/numpy", 1752 "//tensorflow/python:array_ops", 1753 "//tensorflow/python:client_testlib", 1754 "//tensorflow/python:framework_for_generated_wrappers", 1755 "//tensorflow/python:math_ops", 1756 ], 1757 shard_count = 4, 1758 tags = ["no_windows_gpu"], 1759) 1760 1761cuda_py_test( 1762 name = "reduction_ops_test_big", 1763 size = "medium", 1764 srcs = ["reduction_ops_test_big.py"], 1765 additional_deps = [ 1766 "//third_party/py/numpy", 1767 "//tensorflow/python:array_ops", 1768 "//tensorflow/python:client_testlib", 1769 "//tensorflow/python:framework_for_generated_wrappers", 1770 "//tensorflow/python:math_ops", 1771 ], 1772 tags = [ 1773 "manual", 1774 "no_gpu", 1775 "nogpu", 1776 "noguitar", 1777 "notap", 1778 ], 1779) 1780 1781cuda_py_test( 1782 name = "relu_op_test", 1783 size = "small", 1784 srcs = ["relu_op_test.py"], 1785 additional_deps = [ 1786 "//third_party/py/numpy", 1787 "//tensorflow/python:client_testlib", 1788 "//tensorflow/python:framework_for_generated_wrappers", 1789 "//tensorflow/python:gradients", 1790 "//tensorflow/python:nn_grad", 1791 "//tensorflow/python:nn_ops", 1792 "//tensorflow/python:random_ops", 1793 "//tensorflow/python:training", 1794 "//tensorflow/python:variables", 1795 ], 1796) 1797 1798cuda_py_test( 1799 name = "reshape_op_test", 1800 size = "small", 1801 srcs = ["reshape_op_test.py"], 1802 additional_deps = [ 1803 "//third_party/py/numpy", 1804 "//tensorflow/python:array_ops", 1805 "//tensorflow/python:client_testlib", 1806 "//tensorflow/python:framework_for_generated_wrappers", 1807 ], 1808) 1809 1810cuda_py_test( 1811 name = "reverse_sequence_op_test", 1812 size = "small", 1813 srcs = ["reverse_sequence_op_test.py"], 1814 additional_deps = [ 1815 "//third_party/py/numpy", 1816 "//tensorflow/python:array_ops", 1817 "//tensorflow/python:client_testlib", 1818 "//tensorflow/python:framework_for_generated_wrappers", 1819 ], 1820) 1821 1822cuda_py_test( 1823 name = "compare_and_bitpack_op_test", 1824 size = "small", 1825 srcs = ["compare_and_bitpack_op_test.py"], 1826 additional_deps = [ 1827 "//third_party/py/numpy", 1828 "//tensorflow/python:math_ops", 1829 "//tensorflow/python:client_testlib", 1830 "//tensorflow/python:framework_for_generated_wrappers", 1831 ], 1832) 1833 1834cuda_py_test( 1835 name = "scalar_test", 1836 size = "small", 1837 srcs = ["scalar_test.py"], 1838 additional_deps = [ 1839 "//third_party/py/numpy", 1840 "//tensorflow/python:array_ops", 1841 "//tensorflow/python:client_testlib", 1842 "//tensorflow/python:framework_for_generated_wrappers", 1843 "//tensorflow/python:io_ops_gen", 1844 "//tensorflow/python:math_ops", 1845 "//tensorflow/python:nn_grad", 1846 "//tensorflow/python:platform", 1847 "//tensorflow/python:random_ops", 1848 "//tensorflow/python:sparse_ops", 1849 ], 1850) 1851 1852cuda_py_test( 1853 name = "scan_ops_test", 1854 size = "medium", 1855 srcs = ["scan_ops_test.py"], 1856 additional_deps = [ 1857 "//third_party/py/numpy", 1858 "//tensorflow/python:client_testlib", 1859 "//tensorflow/python:errors", 1860 "//tensorflow/python:framework_for_generated_wrappers", 1861 "//tensorflow/python:math_ops", 1862 ], 1863) 1864 1865cuda_py_test( 1866 name = "session_ops_test", 1867 size = "small", 1868 srcs = ["session_ops_test.py"], 1869 additional_deps = [ 1870 "//tensorflow/python:client_testlib", 1871 "//tensorflow/python:framework_for_generated_wrappers", 1872 "//tensorflow/python:math_ops", 1873 "//tensorflow/python:session_ops", 1874 ], 1875) 1876 1877cuda_py_test( 1878 name = "shape_ops_test", 1879 size = "medium", 1880 srcs = ["shape_ops_test.py"], 1881 additional_deps = [ 1882 "//third_party/py/numpy", 1883 "//tensorflow/core:protos_all_py", 1884 "//tensorflow/python:array_ops", 1885 "//tensorflow/python:client_testlib", 1886 "//tensorflow/python:errors", 1887 "//tensorflow/python:framework", 1888 "//tensorflow/python:framework_for_generated_wrappers", 1889 "//tensorflow/python:gradients", 1890 ], 1891) 1892 1893cuda_py_test( 1894 name = "softmax_op_test", 1895 size = "small", 1896 srcs = ["softmax_op_test.py"], 1897 additional_deps = [ 1898 "//third_party/py/numpy", 1899 "//tensorflow/python:array_ops", 1900 "//tensorflow/python:client_testlib", 1901 "//tensorflow/python:errors", 1902 "//tensorflow/python:framework_for_generated_wrappers", 1903 "//tensorflow/python:nn_ops", 1904 ], 1905) 1906 1907cuda_py_test( 1908 name = "softplus_op_test", 1909 size = "small", 1910 srcs = ["softplus_op_test.py"], 1911 additional_deps = [ 1912 "//third_party/py/numpy", 1913 "//tensorflow/python:client_testlib", 1914 "//tensorflow/python:framework_for_generated_wrappers", 1915 "//tensorflow/python:nn_grad", 1916 "//tensorflow/python:nn_ops", 1917 ], 1918) 1919 1920cuda_py_test( 1921 name = "softsign_op_test", 1922 size = "small", 1923 srcs = ["softsign_op_test.py"], 1924 additional_deps = [ 1925 "//third_party/py/numpy", 1926 "//tensorflow/python:client_testlib", 1927 "//tensorflow/python:framework_for_generated_wrappers", 1928 "//tensorflow/python:nn_grad", 1929 "//tensorflow/python:nn_ops", 1930 ], 1931) 1932 1933cuda_py_test( 1934 name = "spacetobatch_op_test", 1935 size = "small", 1936 srcs = ["spacetobatch_op_test.py"], 1937 additional_deps = [ 1938 "//third_party/py/numpy", 1939 "//tensorflow/python:array_ops", 1940 "//tensorflow/python:array_ops_gen", 1941 "//tensorflow/python:client_testlib", 1942 "//tensorflow/python:framework", 1943 "//tensorflow/python:framework_for_generated_wrappers", 1944 "//tensorflow/python:math_ops", 1945 ], 1946) 1947 1948cuda_py_test( 1949 name = "spacetodepth_op_test", 1950 size = "medium", 1951 srcs = ["spacetodepth_op_test.py"], 1952 additional_deps = [ 1953 "//third_party/py/numpy", 1954 "//tensorflow/python:array_ops", 1955 "//tensorflow/python:client_testlib", 1956 "//tensorflow/python:framework_for_generated_wrappers", 1957 "//tensorflow/python:math_ops", 1958 ], 1959) 1960 1961tf_py_test( 1962 name = "sparse_serialization_ops_test", 1963 size = "small", 1964 srcs = ["sparse_serialization_ops_test.py"], 1965 additional_deps = [ 1966 "//third_party/py/numpy", 1967 "//tensorflow/python:array_ops", 1968 "//tensorflow/python:client_testlib", 1969 "//tensorflow/python:framework", 1970 "//tensorflow/python:framework_for_generated_wrappers", 1971 "//tensorflow/python:sparse_ops", 1972 ], 1973) 1974 1975tf_py_test( 1976 name = "sparse_tensors_map_ops_test", 1977 size = "small", 1978 srcs = ["sparse_tensors_map_ops_test.py"], 1979 additional_deps = [ 1980 "//third_party/py/numpy", 1981 "//tensorflow/python:array_ops", 1982 "//tensorflow/python:client", 1983 "//tensorflow/python:client_testlib", 1984 "//tensorflow/python:framework", 1985 "//tensorflow/python:framework_for_generated_wrappers", 1986 "//tensorflow/python:sparse_ops", 1987 "//tensorflow/python:variables", 1988 ], 1989) 1990 1991cuda_py_test( 1992 name = "sparse_tensor_dense_matmul_grad_test", 1993 size = "small", 1994 srcs = ["sparse_tensor_dense_matmul_grad_test.py"], 1995 additional_deps = [ 1996 "//third_party/py/numpy", 1997 "//tensorflow/python:client_testlib", 1998 "//tensorflow/python:framework", 1999 "//tensorflow/python:framework_for_generated_wrappers", 2000 "//tensorflow/python:sparse_grad", 2001 "//tensorflow/python:sparse_ops", 2002 ], 2003) 2004 2005cuda_py_test( 2006 name = "sparse_xent_op_test", 2007 size = "small", 2008 srcs = ["sparse_xent_op_test.py"], 2009 additional_deps = [ 2010 "//third_party/py/numpy", 2011 "//tensorflow/core:protos_all_py", 2012 "//tensorflow/python:array_ops", 2013 "//tensorflow/python:client", 2014 "//tensorflow/python:client_testlib", 2015 "//tensorflow/python:errors", 2016 "//tensorflow/python:framework_for_generated_wrappers", 2017 "//tensorflow/python:gradients", 2018 "//tensorflow/python:math_ops", 2019 "//tensorflow/python:nn_grad", 2020 "//tensorflow/python:nn_ops", 2021 "//tensorflow/python:nn_ops_gen", 2022 "//tensorflow/python:platform", 2023 "//tensorflow/python:sparse_ops", 2024 "//tensorflow/python:random_ops", 2025 "//tensorflow/python:variables", 2026 ], 2027) 2028 2029cuda_py_test( 2030 name = "split_op_test", 2031 size = "medium", 2032 srcs = ["split_op_test.py"], 2033 additional_deps = [ 2034 "//third_party/py/numpy", 2035 "//tensorflow/python:array_ops", 2036 "//tensorflow/python:client_testlib", 2037 "//tensorflow/python:framework_for_generated_wrappers", 2038 "//tensorflow/python:gradients", 2039 "//tensorflow/python:math_ops", 2040 ], 2041) 2042 2043cuda_py_test( 2044 name = "stack_ops_test", 2045 size = "small", 2046 srcs = ["stack_ops_test.py"], 2047 additional_deps = [ 2048 "//third_party/py/numpy", 2049 "//tensorflow/python:client_testlib", 2050 "//tensorflow/python:control_flow_ops", 2051 "//tensorflow/python:data_flow_ops_gen", 2052 "//tensorflow/python:errors", 2053 "//tensorflow/python:framework_for_generated_wrappers", 2054 "//tensorflow/python:math_ops", 2055 ], 2056) 2057 2058cuda_py_test( 2059 name = "string_to_hash_bucket_op_test", 2060 size = "small", 2061 srcs = ["string_to_hash_bucket_op_test.py"], 2062 additional_deps = [ 2063 "//tensorflow/python:array_ops", 2064 "//tensorflow/python:client_testlib", 2065 "//tensorflow/python:framework_for_generated_wrappers", 2066 "//tensorflow/python:string_ops", 2067 ], 2068) 2069 2070cuda_py_test( 2071 name = "string_to_number_op_test", 2072 size = "small", 2073 srcs = ["string_to_number_op_test.py"], 2074 additional_deps = [ 2075 "//tensorflow/python:array_ops", 2076 "//tensorflow/python:client_testlib", 2077 "//tensorflow/python:framework_for_generated_wrappers", 2078 "//tensorflow/python:parsing_ops", 2079 ], 2080 tags = ["no_windows"], 2081) 2082 2083cuda_py_test( 2084 name = "summary_audio_op_test", 2085 size = "small", 2086 srcs = ["summary_audio_op_test.py"], 2087 additional_deps = [ 2088 "//third_party/py/numpy", 2089 "//tensorflow/core:protos_all_py", 2090 "//tensorflow/python:client_testlib", 2091 "//tensorflow/python:framework_for_generated_wrappers", 2092 "//tensorflow/python:summary", 2093 ], 2094) 2095 2096cuda_py_test( 2097 name = "summary_image_op_test", 2098 size = "small", 2099 srcs = ["summary_image_op_test.py"], 2100 additional_deps = [ 2101 "//third_party/py/numpy", 2102 "//tensorflow/core:protos_all_py", 2103 "//tensorflow/python:client_testlib", 2104 "//tensorflow/python:framework_for_generated_wrappers", 2105 "//tensorflow/python:image_ops", 2106 "//tensorflow/python:nn_grad", 2107 "//tensorflow/python:summary", 2108 ], 2109) 2110 2111cuda_py_test( 2112 name = "tensor_array_ops_test", 2113 size = "small", 2114 srcs = ["tensor_array_ops_test.py"], 2115 additional_deps = [ 2116 "//third_party/py/numpy", 2117 "//tensorflow/python:array_ops", 2118 "//tensorflow/python:client_testlib", 2119 "//tensorflow/python:control_flow_ops", 2120 "//tensorflow/python:data_flow_ops_gen", 2121 "//tensorflow/python:distributed_framework_test_lib", 2122 "//tensorflow/python:framework_for_generated_wrappers", 2123 "//tensorflow/python:framework_test_lib", 2124 "//tensorflow/python:gradients", 2125 "//tensorflow/python:init_ops", 2126 "//tensorflow/python:math_ops", 2127 "//tensorflow/python:nn_grad", 2128 "//tensorflow/python:training", 2129 "//tensorflow/python:tensor_array_grad", 2130 "//tensorflow/python:tensor_array_ops", 2131 "//tensorflow/python:variables", 2132 "//tensorflow/python:variable_scope", 2133 "//tensorflow/python/eager:backprop", 2134 "//tensorflow/python/eager:context", 2135 ], 2136 flaky = 1, # create_local_cluster sometimes times out. 2137) 2138 2139cuda_py_test( 2140 name = "trace_op_test", 2141 size = "small", 2142 srcs = ["trace_op_test.py"], 2143 additional_deps = [ 2144 "//third_party/py/numpy", 2145 "//tensorflow/python:client_testlib", 2146 "//tensorflow/python:math_ops", 2147 ], 2148 tags = ["no_windows_gpu"], 2149) 2150 2151cuda_py_test( 2152 name = "transpose_op_test", 2153 size = "large", 2154 srcs = ["transpose_op_test.py"], 2155 additional_deps = [ 2156 "//third_party/py/numpy", 2157 "//tensorflow/python:array_ops", 2158 "//tensorflow/python:client_testlib", 2159 "//tensorflow/python:framework_for_generated_wrappers", 2160 ], 2161 shard_count = 2, 2162 tags = [ 2163 "no_gpu", 2164 "no_oss", 2165 ], 2166) 2167 2168cuda_py_test( 2169 name = "unstack_op_test", 2170 size = "small", 2171 srcs = ["unstack_op_test.py"], 2172 additional_deps = [ 2173 "//third_party/py/numpy", 2174 "//tensorflow/python:array_ops", 2175 "//tensorflow/python:client_testlib", 2176 "//tensorflow/python:framework_for_generated_wrappers", 2177 ], 2178) 2179 2180cuda_py_test( 2181 name = "variable_ops_test", 2182 size = "small", 2183 srcs = ["variable_ops_test.py"], 2184 additional_deps = [ 2185 "//third_party/py/numpy", 2186 "//tensorflow/python:array_ops", 2187 "//tensorflow/python:client_testlib", 2188 "//tensorflow/python:errors", 2189 "//tensorflow/python:framework_for_generated_wrappers", 2190 "//tensorflow/python:math_ops", 2191 "//tensorflow/python:state_ops", 2192 "//tensorflow/python:state_ops_gen", 2193 "//tensorflow/python:variables", 2194 ], 2195) 2196 2197cuda_py_test( 2198 name = "xent_op_test", 2199 size = "small", 2200 srcs = ["xent_op_test.py"], 2201 additional_deps = [ 2202 "//third_party/py/numpy", 2203 "//tensorflow/python:client_testlib", 2204 "//tensorflow/python:framework_for_generated_wrappers", 2205 "//tensorflow/python:gradients", 2206 "//tensorflow/python:math_ops", 2207 "//tensorflow/python:nn_grad", 2208 "//tensorflow/python:nn_ops", 2209 "//tensorflow/python:nn_ops_gen", 2210 ], 2211) 2212 2213cuda_py_test( 2214 name = "zero_division_test", 2215 size = "small", 2216 srcs = ["zero_division_test.py"], 2217 additional_deps = [ 2218 "//tensorflow/python:client_testlib", 2219 "//tensorflow/python:errors", 2220 "//tensorflow/python:framework_for_generated_wrappers", 2221 ], 2222) 2223 2224cuda_py_test( 2225 name = "atrous_conv2d_test", 2226 size = "medium", 2227 srcs = ["atrous_conv2d_test.py"], 2228 additional_deps = [ 2229 "//third_party/py/numpy", 2230 "//tensorflow/python:array_ops", 2231 "//tensorflow/python:client_testlib", 2232 "//tensorflow/python:framework_for_generated_wrappers", 2233 "//tensorflow/python:nn", 2234 "//tensorflow/python:nn_grad", 2235 "//tensorflow/python:nn_ops", 2236 ], 2237 shard_count = 2, 2238) 2239 2240cuda_py_test( 2241 name = "atrous_convolution_test", 2242 size = "medium", 2243 srcs = ["atrous_convolution_test.py"], 2244 additional_deps = [ 2245 "//third_party/py/numpy", 2246 "//tensorflow/python:array_ops", 2247 "//tensorflow/python:client_testlib", 2248 "//tensorflow/python:framework_for_generated_wrappers", 2249 "//tensorflow/python:nn_grad", 2250 "//tensorflow/python:nn_ops", 2251 ], 2252 tags = ["manual"], 2253) 2254 2255cuda_py_test( 2256 name = "pool_test", 2257 size = "medium", 2258 srcs = ["pool_test.py"], 2259 additional_deps = [ 2260 "//third_party/py/numpy", 2261 "//tensorflow/python:client_testlib", 2262 "//tensorflow/python:framework_for_generated_wrappers", 2263 "//tensorflow/python:nn_grad", 2264 "//tensorflow/python:nn_ops", 2265 ], 2266) 2267 2268cuda_py_test( 2269 name = "conv2d_backprop_filter_grad_test", 2270 size = "medium", 2271 srcs = ["conv2d_backprop_filter_grad_test.py"], 2272 additional_deps = [ 2273 "//third_party/py/numpy", 2274 "//tensorflow/python:array_ops", 2275 "//tensorflow/python:client_testlib", 2276 "//tensorflow/python:framework_for_generated_wrappers", 2277 "//tensorflow/python:nn_grad", 2278 "//tensorflow/python:nn_ops", 2279 ], 2280) 2281 2282cuda_py_test( 2283 name = "conv3d_transpose_test", 2284 size = "medium", 2285 srcs = ["conv3d_transpose_test.py"], 2286 additional_deps = [ 2287 "//third_party/py/numpy", 2288 "//tensorflow/python:client_testlib", 2289 "//tensorflow/python:framework_for_generated_wrappers", 2290 "//tensorflow/python:nn_grad", 2291 "//tensorflow/python:nn_ops", 2292 ], 2293) 2294 2295cuda_py_test( 2296 name = "conv_ops_test", 2297 size = "large", 2298 srcs = ["conv_ops_test.py"], 2299 additional_deps = [ 2300 "//third_party/py/numpy", 2301 "//tensorflow/contrib/layers:layers_py", 2302 "//tensorflow/python:array_ops", 2303 "//tensorflow/python:client", 2304 "//tensorflow/python:client_testlib", 2305 "//tensorflow/python:errors", 2306 "//tensorflow/python:framework_for_generated_wrappers", 2307 "//tensorflow/python:framework_test_lib", 2308 "//tensorflow/python:nn", 2309 "//tensorflow/python:nn_grad", 2310 "//tensorflow/python:nn_ops", 2311 "//tensorflow/python:platform", 2312 "//tensorflow/python:random_ops", 2313 "//tensorflow/python:variables", 2314 ], 2315 shard_count = 4, 2316 tags = ["no_windows"], 2317) 2318 2319cuda_py_test( 2320 name = "depthwise_conv_op_test", 2321 size = "medium", # http://b/30603882 2322 srcs = ["depthwise_conv_op_test.py"], 2323 additional_deps = [ 2324 "//third_party/py/numpy", 2325 "//tensorflow/python:client_testlib", 2326 "//tensorflow/python:framework_for_generated_wrappers", 2327 "//tensorflow/python:array_ops", 2328 "//tensorflow/python:nn", 2329 "//tensorflow/python:nn_grad", 2330 "//tensorflow/python:nn_ops", 2331 ], 2332) 2333 2334tf_py_test( 2335 name = "neon_depthwise_conv_op_test", 2336 size = "medium", 2337 srcs = ["neon_depthwise_conv_op_test.py"], 2338 additional_deps = [ 2339 "//third_party/py/numpy", 2340 "//tensorflow/python:client_testlib", 2341 "//tensorflow/python:framework_for_generated_wrappers", 2342 "//tensorflow/python:array_ops", 2343 "//tensorflow/python:nn", 2344 "//tensorflow/python:nn_grad", 2345 "//tensorflow/python:nn_ops", 2346 ], 2347 tags = ["no_windows"], 2348) 2349 2350cuda_py_test( 2351 name = "division_future_test", 2352 size = "medium", 2353 srcs = ["division_future_test.py"], 2354 additional_deps = [ 2355 "//third_party/py/numpy", 2356 "//tensorflow/python:client_testlib", 2357 "//tensorflow/python:framework_for_generated_wrappers", 2358 ], 2359 tags = ["manual"], 2360) 2361 2362cuda_py_test( 2363 name = "dct_ops_test", 2364 srcs = ["dct_ops_test.py"], 2365 additional_deps = [ 2366 "//third_party/py/numpy", 2367 "//tensorflow/python:client_testlib", 2368 "//tensorflow/python:framework_for_generated_wrappers", 2369 "//tensorflow/python:spectral_ops", 2370 "//tensorflow/python:spectral_ops_test_util", 2371 ], 2372) 2373 2374cuda_py_test( 2375 name = "fft_ops_test", 2376 size = "large", 2377 srcs = ["fft_ops_test.py"], 2378 additional_deps = [ 2379 "//third_party/py/numpy", 2380 "//tensorflow/python:client_testlib", 2381 "//tensorflow/python:framework_for_generated_wrappers", 2382 "//tensorflow/python:math_ops", 2383 "//tensorflow/python:spectral_ops", 2384 "//tensorflow/python:spectral_ops_test_util", 2385 ], 2386 shard_count = 3, 2387) 2388 2389cuda_py_test( 2390 name = "pooling_ops_3d_test", 2391 size = "medium", 2392 srcs = ["pooling_ops_3d_test.py"], 2393 additional_deps = [ 2394 "//third_party/py/numpy", 2395 "//tensorflow/python:client_testlib", 2396 "//tensorflow/python:framework_for_generated_wrappers", 2397 "//tensorflow/python:nn_grad", 2398 "//tensorflow/python:nn_ops", 2399 ], 2400) 2401 2402cuda_py_test( 2403 name = "pooling_ops_test", 2404 size = "medium", 2405 srcs = ["pooling_ops_test.py"], 2406 additional_deps = [ 2407 "//third_party/py/numpy", 2408 "//tensorflow/python:array_ops", 2409 "//tensorflow/python:client_testlib", 2410 "//tensorflow/python:errors", 2411 "//tensorflow/python:framework_for_generated_wrappers", 2412 "//tensorflow/python:framework_test_lib", 2413 "//tensorflow/python:nn_grad", 2414 "//tensorflow/python:nn_ops", 2415 "//tensorflow/python:nn_ops_gen", 2416 ], 2417 shard_count = 4, 2418) 2419 2420cuda_py_test( 2421 name = "rnn_test", 2422 size = "medium", 2423 srcs = ["rnn_test.py"], 2424 additional_deps = [ 2425 "//third_party/py/numpy", 2426 "//tensorflow/contrib/rnn:rnn_py", 2427 "//tensorflow/core:protos_all_py", 2428 "//tensorflow/python:array_ops", 2429 "//tensorflow/python:client", 2430 "//tensorflow/python:client_testlib", 2431 "//tensorflow/python:control_flow_ops", 2432 "//tensorflow/python:data_flow_grad", 2433 "//tensorflow/python:framework_for_generated_wrappers", 2434 "//tensorflow/python:framework_test_lib", 2435 "//tensorflow/python:gradients", 2436 "//tensorflow/python:init_ops", 2437 "//tensorflow/python:nn_grad", 2438 "//tensorflow/python:rnn", 2439 "//tensorflow/python:rnn_cell", 2440 "//tensorflow/python:sparse_grad", 2441 "//tensorflow/python:tensor_array_grad", 2442 "//tensorflow/python:tensor_array_ops", 2443 "//tensorflow/python:variables", 2444 "//tensorflow/python/eager:context", 2445 ], 2446 shard_count = 10, 2447 tags = ["no_windows"], 2448) 2449 2450cuda_py_test( 2451 name = "scatter_ops_test", 2452 size = "large", # NOTE: This is not run by default. 2453 srcs = ["scatter_ops_test.py"], 2454 additional_deps = [ 2455 "//third_party/py/numpy", 2456 "//tensorflow/python:client_testlib", 2457 "//tensorflow/python:framework_for_generated_wrappers", 2458 "//tensorflow/python:state_ops", 2459 "//tensorflow/python:variables", 2460 ], 2461) 2462 2463cuda_py_test( 2464 name = "slice_op_test", 2465 size = "large", 2466 srcs = ["slice_op_test.py"], 2467 additional_deps = [ 2468 "//third_party/py/numpy", 2469 "//tensorflow/python:array_ops", 2470 "//tensorflow/python:client_testlib", 2471 "//tensorflow/python:errors", 2472 "//tensorflow/python:framework_for_generated_wrappers", 2473 "//tensorflow/python:gradients", 2474 ], 2475) 2476 2477cuda_py_test( 2478 name = "sparse_matmul_op_test", 2479 size = "medium", 2480 srcs = ["sparse_matmul_op_test.py"], 2481 additional_deps = [ 2482 "//third_party/py/numpy", 2483 "//tensorflow/python:client_testlib", 2484 "//tensorflow/python:framework_for_generated_wrappers", 2485 "//tensorflow/python:math_ops", 2486 ], 2487 tags = ["no_windows"], 2488) 2489 2490cuda_py_test( 2491 name = "sparse_ops_test", 2492 size = "medium", 2493 srcs = ["sparse_ops_test.py"], 2494 additional_deps = [ 2495 "//third_party/py/numpy", 2496 "//tensorflow/python:array_ops", 2497 "//tensorflow/python:client_testlib", 2498 "//tensorflow/python:framework", 2499 "//tensorflow/python:framework_for_generated_wrappers", 2500 "//tensorflow/python:framework_test_lib", 2501 "//tensorflow/python:nn_ops", 2502 "//tensorflow/python:platform_test", 2503 "//tensorflow/python:sparse_grad", 2504 "//tensorflow/python:sparse_ops", 2505 ], 2506 shard_count = 5, 2507 tags = ["noasan"], 2508) 2509 2510cuda_py_test( 2511 name = "sparse_tensor_dense_matmul_op_test", 2512 size = "medium", 2513 srcs = ["sparse_tensor_dense_matmul_op_test.py"], 2514 additional_deps = [ 2515 "//third_party/py/numpy", 2516 "//tensorflow/core:protos_all_py", 2517 "//tensorflow/python:array_ops", 2518 "//tensorflow/python:client", 2519 "//tensorflow/python:client_testlib", 2520 "//tensorflow/python:control_flow_ops", 2521 "//tensorflow/python:framework", 2522 "//tensorflow/python:framework_for_generated_wrappers", 2523 "//tensorflow/python:math_ops", 2524 "//tensorflow/python:platform", 2525 "//tensorflow/python:sparse_ops", 2526 ], 2527) 2528 2529# TODO(gpapan): Revisit the gradient of extract_image_patches_op to resolve 2530# http://b/31080670. 2531cuda_py_test( 2532 name = "extract_image_patches_grad_test", 2533 size = "medium", 2534 srcs = ["extract_image_patches_grad_test.py"], 2535 additional_deps = [ 2536 "//third_party/py/numpy", 2537 "//tensorflow/python:array_ops", 2538 "//tensorflow/python:client_testlib", 2539 "//tensorflow/python:framework", 2540 "//tensorflow/python:framework_for_generated_wrappers", 2541 ], 2542 tags = ["notap"], # http://b/31080670 2543) 2544 2545cuda_py_test( 2546 name = "stage_op_test", 2547 size = "medium", 2548 srcs = ["stage_op_test.py"], 2549 additional_deps = [ 2550 "//tensorflow/python:array_ops", 2551 "//tensorflow/python:client_testlib", 2552 "//tensorflow/python:framework_for_generated_wrappers", 2553 "//tensorflow/python:math_ops", 2554 "//tensorflow/python:util", 2555 "//tensorflow/python:data_flow_ops", 2556 ], 2557) 2558 2559cuda_py_test( 2560 name = "map_stage_op_test", 2561 size = "medium", 2562 srcs = ["map_stage_op_test.py"], 2563 additional_deps = [ 2564 "//tensorflow/python:array_ops", 2565 "//tensorflow/python:client_testlib", 2566 "//tensorflow/python:framework_for_generated_wrappers", 2567 "//tensorflow/python:math_ops", 2568 "//tensorflow/python:util", 2569 "//tensorflow/python:data_flow_ops", 2570 ], 2571) 2572 2573cuda_py_test( 2574 name = "concat_op_test", 2575 size = "medium", 2576 srcs = ["concat_op_test.py"], 2577 additional_deps = [ 2578 "//third_party/py/numpy", 2579 "//tensorflow/python:array_ops", 2580 "//tensorflow/python:array_ops_gen", 2581 "//tensorflow/python:client_testlib", 2582 "//tensorflow/python:errors", 2583 "//tensorflow/python:framework_for_generated_wrappers", 2584 "//tensorflow/python:gradients", 2585 "//tensorflow/python:math_ops", 2586 "//tensorflow/python:variables", 2587 ], 2588) 2589 2590cuda_py_test( 2591 name = "large_concat_op_test", 2592 size = "medium", 2593 srcs = ["large_concat_op_test.py"], 2594 additional_deps = [ 2595 "//tensorflow/python:array_ops", 2596 "//tensorflow/python:client_testlib", 2597 "//tensorflow/python:framework_for_generated_wrappers", 2598 ], 2599 tags = [ 2600 "nomsan", 2601 "notsan", 2602 ], 2603) 2604 2605cuda_py_test( 2606 name = "conv_ops_3d_test", 2607 size = "medium", 2608 srcs = ["conv_ops_3d_test.py"], 2609 additional_deps = [ 2610 "//tensorflow/python:client_testlib", 2611 "//tensorflow/python:framework_for_generated_wrappers", 2612 "//tensorflow/python:nn_grad", 2613 "//tensorflow/python:nn_ops", 2614 ], 2615 shard_count = 20, 2616) 2617 2618cuda_py_test( 2619 name = "cwise_ops_test", 2620 size = "medium", 2621 srcs = ["cwise_ops_test.py"], 2622 additional_deps = [ 2623 "//third_party/py/numpy", 2624 "//tensorflow/python:array_ops", 2625 "//tensorflow/python:client_testlib", 2626 "//tensorflow/python:framework", 2627 "//tensorflow/python:framework_for_generated_wrappers", 2628 "//tensorflow/python:gradients", 2629 "//tensorflow/python:math_ops", 2630 "//tensorflow/python:math_ops_gen", 2631 "//tensorflow/python:nn_grad", 2632 "//tensorflow/python:platform", 2633 "//tensorflow/python:variables", 2634 ], 2635 shard_count = 50, 2636 tags = [ 2637 "manual", 2638 "notap", # b/30226163 2639 ], 2640) 2641 2642cuda_py_test( 2643 name = "embedding_ops_test", 2644 size = "medium", 2645 srcs = ["embedding_ops_test.py"], 2646 additional_deps = [ 2647 "//third_party/py/numpy", 2648 "//tensorflow/python:array_ops", 2649 "//tensorflow/python:client_testlib", 2650 "//tensorflow/python:data_flow_ops", 2651 "//tensorflow/python:embedding_ops", 2652 "//tensorflow/python:framework", 2653 "//tensorflow/python:framework_for_generated_wrappers", 2654 "//tensorflow/python:linalg_ops", 2655 "//tensorflow/python:math_ops", 2656 "//tensorflow/python:partitioned_variables", 2657 "//tensorflow/python:platform", 2658 "//tensorflow/python:state_ops", 2659 "//tensorflow/python:util", 2660 "//tensorflow/python:variable_scope", 2661 "//tensorflow/python:variables", 2662 ], 2663 shard_count = 20, 2664) 2665 2666cuda_py_test( 2667 name = "linalg_grad_test", 2668 size = "medium", 2669 srcs = ["linalg_grad_test.py"], 2670 additional_deps = [ 2671 "//third_party/py/numpy", 2672 "//tensorflow/python:array_ops", 2673 "//tensorflow/python:client_testlib", 2674 "//tensorflow/python:framework_for_generated_wrappers", 2675 "//tensorflow/python:gradients", 2676 "//tensorflow/python:linalg_ops", 2677 "//tensorflow/python:math_ops", 2678 "//tensorflow/python/ops/linalg", 2679 ], 2680 shard_count = 20, 2681) 2682 2683cuda_py_test( 2684 name = "matrix_band_part_op_test", 2685 size = "medium", 2686 srcs = ["matrix_band_part_op_test.py"], 2687 additional_deps = [ 2688 "//third_party/py/numpy", 2689 "//tensorflow/python:array_ops", 2690 "//tensorflow/python:client_testlib", 2691 "//tensorflow/python:framework_for_generated_wrappers", 2692 ], 2693 shard_count = 20, 2694) 2695 2696cuda_py_test( 2697 name = "self_adjoint_eig_op_test", 2698 size = "medium", 2699 srcs = ["self_adjoint_eig_op_test.py"], 2700 additional_deps = [ 2701 "//third_party/py/numpy", 2702 "//tensorflow/python:array_ops", 2703 "//tensorflow/python:client_testlib", 2704 "//tensorflow/python:framework_for_generated_wrappers", 2705 "//tensorflow/python:linalg_ops", 2706 "//tensorflow/python:math_ops", 2707 ], 2708 shard_count = 20, 2709) 2710 2711cuda_py_test( 2712 name = "qr_op_test", 2713 size = "medium", 2714 srcs = ["qr_op_test.py"], 2715 additional_deps = [ 2716 "//third_party/py/numpy", 2717 "//tensorflow/python:array_ops", 2718 "//tensorflow/python:client_testlib", 2719 "//tensorflow/python:framework_for_generated_wrappers", 2720 "//tensorflow/python:linalg_ops", 2721 "//tensorflow/python:math_ops", 2722 ], 2723 shard_count = 20, 2724) 2725 2726cuda_py_test( 2727 name = "svd_op_test", 2728 size = "medium", 2729 srcs = ["svd_op_test.py"], 2730 additional_deps = [ 2731 "//third_party/py/numpy", 2732 "//tensorflow/python:array_ops", 2733 "//tensorflow/python:client_testlib", 2734 "//tensorflow/python:framework_for_generated_wrappers", 2735 "//tensorflow/python:linalg_ops", 2736 "//tensorflow/python:math_ops", 2737 ], 2738 shard_count = 20, 2739) 2740 2741cuda_py_test( 2742 name = "norm_op_test", 2743 size = "medium", 2744 srcs = ["norm_op_test.py"], 2745 additional_deps = [ 2746 "//third_party/py/numpy", 2747 "//tensorflow/python:array_ops", 2748 "//tensorflow/python:client_testlib", 2749 "//tensorflow/python:framework_for_generated_wrappers", 2750 "//tensorflow/python:linalg_ops", 2751 ], 2752 shard_count = 20, 2753 tags = ["no_windows_gpu"], 2754) 2755 2756cuda_py_test( 2757 name = "tensordot_op_test", 2758 size = "medium", 2759 srcs = ["tensordot_op_test.py"], 2760 additional_deps = [ 2761 "//third_party/py/numpy", 2762 "//tensorflow/python:array_ops", 2763 "//tensorflow/python:client_testlib", 2764 "//tensorflow/python:errors", 2765 "//tensorflow/python:framework_for_generated_wrappers", 2766 "//tensorflow/python:math_ops", 2767 ], 2768 shard_count = 20, 2769) 2770 2771sycl_py_test( 2772 name = "basic_gpu_test", 2773 size = "small", 2774 srcs = ["basic_gpu_test.py"], 2775 additional_deps = [ 2776 "//third_party/py/numpy", 2777 "//tensorflow/python:array_ops_gen", 2778 "//tensorflow/python:client_testlib", 2779 "//tensorflow/python:framework_for_generated_wrappers", 2780 "//tensorflow/python:math_ops", 2781 "//tensorflow/python:math_ops_gen", 2782 ], 2783) 2784 2785tf_py_test( 2786 name = "sets_test", 2787 size = "small", 2788 srcs = ["sets_test.py"], 2789 additional_deps = [ 2790 "//third_party/py/numpy", 2791 "//tensorflow/python:errors", 2792 "//tensorflow/python:framework", 2793 "//tensorflow/python:framework_for_generated_wrappers", 2794 "//tensorflow/python:framework_test_lib", 2795 "//tensorflow/python:math_ops", 2796 "//tensorflow/python:platform_test", 2797 "//tensorflow/python:sets", 2798 "//tensorflow/python:sparse_ops", 2799 ], 2800) 2801 2802tf_py_test( 2803 name = "weights_broadcast_test", 2804 size = "small", 2805 srcs = ["weights_broadcast_test.py"], 2806 additional_deps = [ 2807 "//third_party/py/numpy", 2808 "//tensorflow/python:array_ops", 2809 "//tensorflow/python:client_testlib", 2810 "//tensorflow/python:errors", 2811 "//tensorflow/python:framework", 2812 "//tensorflow/python:framework_for_generated_wrappers", 2813 "//tensorflow/python:framework_test_lib", 2814 "//tensorflow/python:platform", 2815 "//tensorflow/python:weights_broadcast_ops", 2816 ], 2817 shard_count = 3, 2818) 2819 2820tf_py_test( 2821 name = "metrics_test", 2822 size = "medium", 2823 srcs = ["metrics_test.py"], 2824 additional_deps = [ 2825 "//third_party/py/numpy", 2826 "//tensorflow/python:array_ops", 2827 "//tensorflow/python:client_testlib", 2828 "//tensorflow/python:data_flow_grad", 2829 "//tensorflow/python:data_flow_ops", 2830 "//tensorflow/python:errors", 2831 "//tensorflow/python:framework", 2832 "//tensorflow/python:framework_for_generated_wrappers", 2833 "//tensorflow/python:metrics", 2834 "//tensorflow/python:nn_grad", 2835 "//tensorflow/python:random_ops", 2836 "//tensorflow/python:variables", 2837 ], 2838 shard_count = 10, 2839 tags = ["no_windows_gpu"], 2840) 2841 2842tf_py_test( 2843 name = "confusion_matrix_test", 2844 size = "small", 2845 srcs = ["confusion_matrix_test.py"], 2846 additional_deps = [ 2847 "//third_party/py/numpy", 2848 "//tensorflow/python:array_ops", 2849 "//tensorflow/python:client_testlib", 2850 "//tensorflow/python:confusion_matrix", 2851 "//tensorflow/python:framework_for_generated_wrappers", 2852 "//tensorflow/python:math_ops", 2853 "//tensorflow/python:random_ops", 2854 ], 2855) 2856 2857tf_py_test( 2858 name = "bucketize_op_test", 2859 size = "small", 2860 srcs = ["bucketize_op_test.py"], 2861 additional_deps = [ 2862 "//third_party/py/numpy", 2863 "//tensorflow/python:client_testlib", 2864 "//tensorflow/python:framework_for_generated_wrappers", 2865 "//tensorflow/python:math_ops", 2866 ], 2867) 2868 2869tf_py_test( 2870 name = "sparse_cross_op_test", 2871 size = "small", 2872 srcs = ["sparse_cross_op_test.py"], 2873 additional_deps = [ 2874 "//third_party/py/numpy", 2875 "//tensorflow/python:client_testlib", 2876 "//tensorflow/python:framework_for_generated_wrappers", 2877 "//tensorflow/python:sparse_ops", 2878 ], 2879) 2880 2881tf_py_test( 2882 name = "garbage_collection_test", 2883 size = "small", 2884 srcs = ["garbage_collection_test.py"], 2885 additional_deps = [ 2886 "//tensorflow/python/eager:context", 2887 "//tensorflow/python:framework_test_lib", 2888 "//tensorflow/python:resource_variable_ops", 2889 "//tensorflow/python:dtypes", 2890 "//tensorflow/python:tensor_array_ops", 2891 "//tensorflow/python:client_testlib", 2892 ], 2893) 2894 2895filegroup( 2896 name = "all_files", 2897 srcs = glob( 2898 ["**/*"], 2899 exclude = [ 2900 "**/METADATA", 2901 "**/OWNERS", 2902 ], 2903 ), 2904 visibility = ["//tensorflow:__subpackages__"], 2905) 2906