Home | History | Annotate | Download | only in utils
      1 licenses(["notice"])  # Apache 2.0
      2 
      3 exports_files(["LICENSE"])
      4 
      5 load("//tensorflow:tensorflow.bzl", "py_test")
      6 
      7 filegroup(
      8     name = "all_files",
      9     srcs = glob(
     10         ["**/*"],
     11         exclude = [
     12             "**/METADATA",
     13             "**/OWNERS",
     14         ],
     15     ),
     16     visibility = ["//tensorflow:__subpackages__"],
     17 )
     18 
     19 py_library(
     20     name = "utils",
     21     srcs = [
     22         "__init__.py",
     23         "context_managers.py",
     24         "misc.py",
     25         "multiple_dispatch.py",
     26         "printing.py",
     27         "py_func.py",
     28         "tensor_list.py",
     29         "type_check.py",
     30     ],
     31     srcs_version = "PY2AND3",
     32     visibility = ["//tensorflow:__subpackages__"],
     33     deps = [
     34         "//tensorflow/python:script_ops",
     35         "@six_archive//:six",
     36     ],
     37 )
     38 
     39 py_test(
     40     name = "context_managers_test",
     41     srcs = ["context_managers_test.py"],
     42     srcs_version = "PY2AND3",
     43     deps = [
     44         ":utils",
     45         "//tensorflow/python:client_testlib",
     46     ],
     47 )
     48 
     49 py_test(
     50     name = "misc_test",
     51     srcs = ["misc_test.py"],
     52     srcs_version = "PY2AND3",
     53     deps = [
     54         ":utils",
     55         "//tensorflow/python:client_testlib",
     56     ],
     57 )
     58 
     59 py_test(
     60     name = "multiple_dispatch_test",
     61     srcs = ["multiple_dispatch_test.py"],
     62     srcs_version = "PY2AND3",
     63     deps = [
     64         ":utils",
     65         "//tensorflow/python:client_testlib",
     66     ],
     67 )
     68 
     69 py_test(
     70     name = "py_func_test",
     71     srcs = ["py_func_test.py"],
     72     srcs_version = "PY2AND3",
     73     deps = [
     74         ":utils",
     75         "//tensorflow/python:client_testlib",
     76     ],
     77 )
     78 
     79 py_test(
     80     name = "printing_test",
     81     srcs = ["printing_test.py"],
     82     srcs_version = "PY2AND3",
     83     deps = [
     84         ":utils",
     85         "//tensorflow/python:client_testlib",
     86     ],
     87 )
     88 
     89 py_test(
     90     name = "type_check_test",
     91     srcs = ["type_check_test.py"],
     92     srcs_version = "PY2AND3",
     93     deps = [
     94         ":utils",
     95         "//tensorflow/python:client_testlib",
     96     ],
     97 )
     98 
     99 py_test(
    100     name = "tensor_list_test",
    101     srcs = ["tensor_list_test.py"],
    102     srcs_version = "PY2AND3",
    103     deps = [
    104         ":utils",
    105         "//tensorflow/python:client_testlib",
    106         "//tensorflow/python:list_ops",
    107     ],
    108 )
    109