Searched refs:embedding_lookup_v2 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings.py | 194 out = embedding_ops.embedding_lookup_v2(self.embeddings.variables, inputs)
196 out = embedding_ops.embedding_lookup_v2(self.embeddings, inputs)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable_test.py | 286 lambda x: embedding_ops.embedding_lookup_v2(root.v.variables, x))
290 lambda x: embedding_ops.embedding_lookup_v2(root.v.variables[0], x),
413 return embedding_ops.embedding_lookup_v2(sv, ids)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 1601 outputs.append(embedding_ops.embedding_lookup_v2(table, inp))
1630 ragged_lookup = embedding_ops.embedding_lookup_v2(table, inp)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 333 def embedding_lookup_v2(params, ids, max_norm=None, name=None): function
|