| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 140 vocab_size, argument
170 vocab_size, argument
190 vocab_size, argument
650 def _RandomIdsAndWeights(self, batch_size, vocab_size): argument
792 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): argument
|
| D | sparse_ops_test.py | 136 def _AssertResultsSorted(self, output, vocab_size): argument
142 def _AssertResultsNotSorted(self, output, vocab_size): argument
251 def _AssertResultsSorted(self, output, vocab_size): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 552 vocab_size=None, argument
679 vocab_size=None, argument
730 vocab_size=None, argument
1219 vocab_size=None, argument
1438 vocab_size=None, argument
|
| D | sparse_ops.py | 1506 def sparse_to_indicator(sp_input, vocab_size, name=None): argument
1570 def sparse_merge(sp_ids, sp_values, vocab_size, name=None, argument
1668 vocab_size, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 75 Status Init(const string& filename, int64 vocab_size, char delimiter, in Init()
349 Status InitializeTableFromTextFile(const string& filename, int64 vocab_size, in InitializeTableFromTextFile()
|
| D | word2vec_kernels.cc | 284 const int64 vocab_size = w_in.dim_size(0); in Compute() local
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_subclassing_test.py | 152 def __init__(self, vocab_size, embedding_dim, **kwargs): argument
170 def __init__(self, vocab_size, embed_size): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | text_vectorization.py | 709 def __init__(self, vocab_size=None, compute_idf=False): argument
|
| D | text_vectorization_test.py | 1650 vocab_size, argument
|