diff options
author | 2018-10-01 16:07:09 -0700 | |
---|---|---|
committer | 2018-10-01 16:11:58 -0700 | |
commit | 55f561e6740d61b3665594babce4be72ad955bc6 (patch) | |
tree | 483c3a7a233928d03fcbc856c6058e9282d2323a /tensorflow/contrib/tpu | |
parent | 8559bc2c4c7616c5da8b4f7a3e1405c549a6068d (diff) |
Small tweaks to comments and documentation strings.
PiperOrigin-RevId: 215297961
Diffstat (limited to 'tensorflow/contrib/tpu')
-rw-r--r-- | tensorflow/contrib/tpu/ops/tpu_embedding_ops.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/tpu/ops/tpu_embedding_ops.cc b/tensorflow/contrib/tpu/ops/tpu_embedding_ops.cc index 5c27d59f82..ef2f8dd36d 100644 --- a/tensorflow/contrib/tpu/ops/tpu_embedding_ops.cc +++ b/tensorflow/contrib/tpu/ops/tpu_embedding_ops.cc @@ -46,7 +46,7 @@ namespace tensorflow { // 5. TPUEmbeddingActivations, when used with appropriate Python libraries, // enables the automatic differentiation of models that use embeddings. // 6. TPUEmbeddingSendGradients takes a list of Tensors (of the same shapes -// as those returned by TPUEmbeddingReceivActivations) containing gradients +// as those returned by TPUEmbeddingReceiveActivations) containing gradients // to use in updating the embedding tables. // 7. Before saving a checkpoint, use the TPUEmbeddingRetrieve Op to update // the Graph's embedding table Variables from the updated tables in the @@ -147,7 +147,7 @@ parameters that are loaded from a checkpoint before a training loop is executed. %s table_name: Name of this table; must match a name in the - EmbeddingLayerConfiguration proto (overrides table_id). + TPUEmbeddingConfiguration proto (overrides table_id). num_shards: Number of shards into which the embedding tables are divided. shard_id: Identifier of shard for this operation. table_id: Index of this table in the EmbeddingLayerConfiguration proto @@ -283,7 +283,7 @@ the correct embedding table configuration. For example, this op is used to retrieve updated parameters before saving a checkpoint. %s table_name: Name of this table; must match a name in the - EmbeddingLayerConfiguration proto (overrides table_id). + TPUEmbeddingConfiguration proto (overrides table_id). num_shards: Number of shards into which the embedding tables are divided. shard_id: Identifier of shard for this operation. table_id: Index of this table in the EmbeddingLayerConfiguration proto |