diff options
author | Daryl Ng <darylng@google.com> | 2018-09-27 20:10:31 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 20:13:42 -0700 |
commit | d377fdee3a5e266ac330a6742c15ece8e7ed8aa0 (patch) | |
tree | 47747c896b3ea76d605cdff8bc07a9196b85fb65 /tensorflow/contrib/tpu | |
parent | a309e136dcfdd13dc8e8eb7570b6c5945bb6f967 (diff) |
Adding to tpu_lib depenencies to optimization_parameters_py, tpu_embedding_configuration_py, and tpu_embedding_output_layout_py.
PiperOrigin-RevId: 214879168
Diffstat (limited to 'tensorflow/contrib/tpu')
-rw-r--r-- | tensorflow/contrib/tpu/BUILD | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/tpu/BUILD b/tensorflow/contrib/tpu/BUILD index ac38612603..e9aa037634 100644 --- a/tensorflow/contrib/tpu/BUILD +++ b/tensorflow/contrib/tpu/BUILD @@ -252,7 +252,10 @@ py_library( ":tpu_py", "//tensorflow/contrib/cluster_resolver:tpu_cluster_resolver_py", "//tensorflow/contrib/tpu/proto:compilation_result_proto_py", + "//tensorflow/contrib/tpu/proto:optimization_parameters_proto_py", "//tensorflow/contrib/tpu/proto:topology_proto_py", + "//tensorflow/contrib/tpu/proto:tpu_embedding_configuration_proto_py", + "//tensorflow/contrib/tpu/proto:tpu_embedding_output_layout_proto_py", "//tensorflow/core:protos_all_py", "//tensorflow/python:array_ops", "//tensorflow/python:control_flow_ops", |