blob: abcd521a265c26e64ec1dfd144e74ea7d3e5bb8c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# Tools for running distributed benchmarks.
licenses(["notice"]) # Apache 2.0
exports_files(["k8s_tensorflow.py"])
py_library(
name = "k8s_tensorflow_lib",
srcs = ["k8s_tensorflow_lib.py"],
srcs_version = "PY2AND3",
)
py_test(
name = "k8s_tensorflow_test",
srcs = ["k8s_tensorflow_test.py"],
srcs_version = "PY2AND3",
deps = [
":k8s_tensorflow_lib",
"//tensorflow/python:client_testlib",
],
)
|