Searched refs:EmbeddingLookupSparseOpModel (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 32 class EmbeddingLookupSparseOpModel : public SingleOpModel { class
34 EmbeddingLookupSparseOpModel(CombinerType type, in EmbeddingLookupSparseOpModel() function in tflite::__anonb811566d0111::EmbeddingLookupSparseOpModel
88 EmbeddingLookupSparseOpModel m(CombinerType_SUM, {3}, {3, 2}, {2}, {4, 3, 2}); in TEST()
103 EmbeddingLookupSparseOpModel m(CombinerType_MEAN, {3}, {3, 2}, {2}, in TEST()
119 EmbeddingLookupSparseOpModel m(CombinerType_SQRTN, {3}, {3, 2}, {2}, in TEST()
139 EmbeddingLookupSparseOpModel m(CombinerType_SUM, {3}, {3, 3}, {3}, {4, 3, 2}); in TEST()
|