Searched refs:EmbeddingSize (Results 1 – 5 of 5) sorted by relevance
66 EXPECT_EQ(tefe.EmbeddingSize(0), 5000); in TEST_F()68 EXPECT_EQ(tefe.EmbeddingSize(1), 7000); in TEST_F()131 EXPECT_EQ(tefe.EmbeddingSize(0), 1000); in TEST_F()137 EXPECT_GT(tefe.EmbeddingSize(1), 0); in TEST_F()
192 *selection_network_, selection_network_->EmbeddingSize(0)); in LoadModels()204 *sharing_network_, sharing_network_->EmbeddingSize(0)); in LoadModels()242 const int embedding_size = network.EmbeddingSize(0); in InferInternal()373 int embedding_size = selection_network_->EmbeddingSize(0); in SuggestSelectionSymmetrical()
199 int EmbeddingSize(int es_index) const;
66 int EmbeddingSize(int index) const { in EmbeddingSize() function
375 int EmbeddingNetwork::EmbeddingSize(int es_index) const { in EmbeddingSize() function in libtextclassifier::nlp_core::EmbeddingNetwork