diff options
| author |  Daryl Ng <darylng@google.com> | 2018-09-20 11:23:28 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-20 11:26:52 -0700 |
| commit | 30756301bee0de2b1c16a74a710bd7bf29be468d (patch) | |
| tree | aaf5ee628804ff0257967f5d1fe1a0116447c166 /tensorflow/contrib/tpu/BUILD | |
| parent | ae59f459cd1e6bd2f2bdeb3b49cfedf0cdaf51a1 (diff) | |
Moving tpu_embedding_config.proto to tpu_embedding_configuration.proto, refactoring it, adding several new fields and an EmbeddingOutputLayout message to provide experimental support for controlling the embedding output.
PiperOrigin-RevId: 213849572
Diffstat (limited to 'tensorflow/contrib/tpu/BUILD')
| -rw-r--r-- | tensorflow/contrib/tpu/BUILD | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/tpu/BUILD b/tensorflow/contrib/tpu/BUILD index 298ffc1ded..87d00aca05 100644 --- a/tensorflow/contrib/tpu/BUILD +++ b/tensorflow/contrib/tpu/BUILD @@ -80,7 +80,7 @@ tf_gen_op_libs( "tpu_embedding_ops", ], deps = [ - "//tensorflow/contrib/tpu/proto:tpu_embedding_config_proto_cc", + "//tensorflow/contrib/tpu/proto:tpu_embedding_configuration_proto_cc", "//tensorflow/core:lib_proto_parsing", "//tensorflow/core:protos_all_cc", ], @@ -99,7 +99,7 @@ tf_custom_op_library( "ops/tpu_embedding_ops.cc", ], deps = [ - "//tensorflow/contrib/tpu/proto:tpu_embedding_config_proto_cc", + "//tensorflow/contrib/tpu/proto:tpu_embedding_configuration_proto_cc", "//tensorflow/core:lib_proto_parsing", ], ) |