| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 141 vocab_size, argument
171 vocab_size, argument
191 vocab_size, argument
655 def _RandomIdsAndWeights(self, batch_size, vocab_size): argument
797 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/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| D | sentencepiece_tokenizer.py | 123 def vocab_size(self): member in SentencepieceTokenizer
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | index_lookup.py | 288 def vocab_size(self): member in IndexLookup
577 vocab_size=None, argument
|
| D | table_utils.py | 72 def vocab_size(self): member in TableHandler
|
| D | index_lookup_test.py | 1657 def test_combiner_computation(self, data, vocab_size, argument
1781 def test_combiner_computation(self, data, vocab_size, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 648 vocab_size=None, argument
785 vocab_size=None, argument
836 vocab_size=None, argument
1353 vocab_size=None, argument
1572 vocab_size=None, argument
|
| D | sparse_ops.py | 1730 def sparse_to_indicator(sp_input, vocab_size, name=None): argument
1794 def sparse_merge(sp_ids, sp_values, vocab_size, name=None, argument
1892 vocab_size, argument
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | text_classification_transformer_benchmark_test.py | 221 def __init__(self, maxlen, vocab_size, embed_dim): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 78 Status Init(const string& filename, int64 vocab_size, char delimiter, in Init()
356 Status InitializeTableFromTextFile(const string& filename, int64 vocab_size, in InitializeTableFromTextFile()
|
| D | word2vec_kernels.cc | 285 const int64 vocab_size = w_in.dim_size(0); in Compute() local
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_subclassing_test.py | 151 def __init__(self, vocab_size, embedding_dim, **kwargs): argument
169 def __init__(self, vocab_size, embed_size): argument
|