Searched refs:EmbeddingLookupSparseOpModel (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 36 class EmbeddingLookupSparseOpModel : public SingleOpModel { class
38 EmbeddingLookupSparseOpModel(CombinerType type, in EmbeddingLookupSparseOpModel() function in tflite::__anonc8d704f40111::EmbeddingLookupSparseOpModel
93 EmbeddingLookupSparseOpModel m(CombinerType_SUM, {3}, {3, 2}, {2}, {4, 3, 2}); in TEST()
108 EmbeddingLookupSparseOpModel m(CombinerType_MEAN, {3}, {3, 2}, {2}, in TEST()
124 EmbeddingLookupSparseOpModel m(CombinerType_SQRTN, {3}, {3, 2}, {2}, in TEST()
144 EmbeddingLookupSparseOpModel m(CombinerType_SUM, {3}, {3, 3}, {3}, {4, 3, 2}); in TEST()
|