Lines Matching refs:k8s_tensorflow_lib
15 """Tests for tensorflow.tools.dist_test.scripts.k8s_tensorflow_lib."""
22 from tensorflow.tools.dist_test.scripts import k8s_tensorflow_lib
29 config = k8s_tensorflow_lib.GenerateConfig(
40 config = k8s_tensorflow_lib.GenerateConfig(
52 config = k8s_tensorflow_lib.GenerateConfig(
63 config = k8s_tensorflow_lib.GenerateConfig(
75 config = k8s_tensorflow_lib.GenerateConfig(
89 config = k8s_tensorflow_lib.GenerateConfig(
104 config = k8s_tensorflow_lib.GenerateConfig(
120 k8s_tensorflow_lib.WorkerHosts(1, 1234, 'test_prefix'))
123 k8s_tensorflow_lib.WorkerHosts(2, 1234, 'test_prefix'))
128 k8s_tensorflow_lib.PsHosts(2, 1234, 'test_prefix'))
