diff options
author | 2018-01-22 17:16:44 -0800 | |
---|---|---|
committer | 2018-01-22 17:22:31 -0800 | |
commit | 2968447d32bdfd0dd6fafabfcd1aafd6dc261803 (patch) | |
tree | ca1432cc15d217f8b3e1881b4461cd602657382b /tensorflow/python/ops/embedding_ops.py | |
parent | f5356644fd3467afa275ec2155281825b7e927a0 (diff) |
Adding tf_export decorators/calls to TensorFlow functions and constants.
PiperOrigin-RevId: 182862075
Diffstat (limited to 'tensorflow/python/ops/embedding_ops.py')
-rw-r--r-- | tensorflow/python/ops/embedding_ops.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/ops/embedding_ops.py b/tensorflow/python/ops/embedding_ops.py index f4561d1a83..3826585f59 100644 --- a/tensorflow/python/ops/embedding_ops.py +++ b/tensorflow/python/ops/embedding_ops.py @@ -32,6 +32,7 @@ from tensorflow.python.ops import math_ops from tensorflow.python.ops import resource_variable_ops from tensorflow.python.ops import variables from tensorflow.python.platform import tf_logging as logging +from tensorflow.python.util.tf_export import tf_export def _gather(params, ids, name=None): @@ -257,6 +258,7 @@ def _embedding_lookup_and_transform(params, return ret +@tf_export("nn.embedding_lookup") def embedding_lookup( params, ids, @@ -325,6 +327,7 @@ def embedding_lookup( transform_fn=None) +@tf_export("nn.embedding_lookup_sparse") def embedding_lookup_sparse(params, sp_ids, sp_weights, |