From 9fafb44f8723d8f4c9863585c912ed0fbdd0a323 Mon Sep 17 00:00:00 2001 From: Yanan Cao Date: Wed, 17 Oct 2018 13:38:53 -0700 Subject: [PATCH] Fix tpu_feed.py dependency PiperOrigin-RevId: 217578782 --- tensorflow/contrib/tpu/BUILD | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/contrib/tpu/BUILD b/tensorflow/contrib/tpu/BUILD index 401afcb0f44..cb5f644ee83 100644 --- a/tensorflow/contrib/tpu/BUILD +++ b/tensorflow/contrib/tpu/BUILD @@ -69,8 +69,6 @@ py_library( deps = [ ":async_checkpoint", ":tpu_lib", - "//tensorflow/compiler/xla/experimental/xla_sharding", - "//tensorflow/compiler/xla/python_api:xla_shape", "//tensorflow/contrib/training:training_py", "//tensorflow/core:protos_all_py", "//tensorflow/python:array_ops", @@ -258,6 +256,8 @@ py_library( ":datasets", ":profiler", ":tpu_py", + "//tensorflow/compiler/xla/experimental/xla_sharding", + "//tensorflow/compiler/xla/python_api:xla_shape", "//tensorflow/contrib/cluster_resolver:tpu_cluster_resolver_py", "//tensorflow/contrib/tpu/proto:compilation_result_proto_py", "//tensorflow/contrib/tpu/proto:optimization_parameters_proto_py",