Home | History | Annotate | Download | only in cluster_resolver 
      1 # Description: Operations defined for Cluster Resolvers
      2 
      3 load("//tensorflow:tensorflow.bzl", "tf_py_test")
      4 
      5 package(
      6     default_visibility = [
      7         "//tensorflow:__subpackages__",
      8     ],
      9 )
     10 
     11 licenses(["notice"])  # Apache 2.0
     12 
     13 filegroup(
     14     name = "all_files",
     15     srcs = glob(
     16         include = [
     17             "**/*",
     18         ],
     19         exclude = [
     20             "**/METADATA",
     21             "**/OWNERS",
     22         ],
     23     ),
     24 )
     25 
     26 py_library(
     27     name = "cluster_resolver_pip",
     28     srcs = [
     29         "__init__.py",
     30         "python/training/__init__.py",
     31     ],
     32     srcs_version = "PY2AND3",
     33     deps = [
     34         ":cluster_resolver_py",
     35         ":gce_cluster_resolver_py",
     36         ":tpu_cluster_resolver_py",
     37         "//tensorflow/python:util",
     38     ],
     39 )
     40 
     41 py_library(
     42     name = "cluster_resolver_py",
     43     srcs = ["python/training/cluster_resolver.py"],
     44     srcs_version = "PY2AND3",
     45     deps = [
     46         "//tensorflow/python:training",
     47     ],
     48 )
     49 
     50 py_library(
     51     name = "gce_cluster_resolver_py",
     52     srcs = ["python/training/gce_cluster_resolver.py"],
     53     srcs_version = "PY2AND3",
     54     deps = [
     55         ":cluster_resolver_py",
     56         "//tensorflow/python:training",
     57     ],
     58 )
     59 
     60 py_library(
     61     name = "tpu_cluster_resolver_py",
     62     srcs = ["python/training/tpu_cluster_resolver.py"],
     63     srcs_version = "PY2AND3",
     64     deps = [
     65         ":cluster_resolver_py",
     66         "//tensorflow/python:training",
     67     ],
     68 )
     69 
     70 tf_py_test(
     71     name = "cluster_resolver_py_test",
     72     srcs = ["python/training/cluster_resolver_test.py"],
     73     additional_deps = [
     74         ":cluster_resolver_py",
     75         "//tensorflow/python:client_testlib",
     76         "//tensorflow/python:framework_for_generated_wrappers",
     77         "//tensorflow/python:framework_test_lib",
     78         "//tensorflow/python:platform_test",
     79         "//tensorflow/python:training",
     80     ],
     81     main = "python/training/cluster_resolver_test.py",
     82 )
     83 
     84 tf_py_test(
     85     name = "gce_cluster_resolver_py_test",
     86     size = "small",
     87     srcs = ["python/training/gce_cluster_resolver_test.py"],
     88     additional_deps = [
     89         ":cluster_resolver_py",
     90         ":gce_cluster_resolver_py",
     91         "//tensorflow/python:client_testlib",
     92         "//tensorflow/python:framework_for_generated_wrappers",
     93         "//tensorflow/python:framework_test_lib",
     94         "//tensorflow/python:platform_test",
     95         "//tensorflow/python:training",
     96     ],
     97     main = "python/training/gce_cluster_resolver_test.py",
     98 )
     99 
    100 tf_py_test(
    101     name = "tpu_cluster_resolver_py_test",
    102     size = "small",
    103     srcs = ["python/training/tpu_cluster_resolver_test.py"],
    104     additional_deps = [
    105         ":tpu_cluster_resolver_py",
    106         "//tensorflow/python:client_testlib",
    107         "//tensorflow/python:framework_for_generated_wrappers",
    108         "//tensorflow/python:framework_test_lib",
    109         "//tensorflow/python:platform_test",
    110         "//tensorflow/python:training",
    111     ],
    112     main = "python/training/tpu_cluster_resolver_test.py",
    113 )
    114