Searched refs:embed_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | encoders.py | 34 embed_dim, argument
68 if not vocab_size or not embed_dim:
73 'embeddings', shape=[vocab_size, embed_dim],
92 embed_dim=None, argument
126 if not (reuse or (vocab_size and embed_dim)):
129 vocab_size, embed_dim))
132 shape = [vocab_size, embed_dim]
133 if reuse and vocab_size is None or embed_dim is None:
|
| D | embedding_ops_test.py | 47 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): argument
49 assert embed_dim > 0
57 shape=[vocab_size, embed_dim],
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 792 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): argument
794 assert embed_dim > 0
802 shape=[vocab_size, embed_dim],
|