Searched refs:EmbeddingDims (Results 1 – 2 of 2) sorted by relevance
67 EXPECT_EQ(tefe.EmbeddingDims(0), 16); in TEST_F()69 EXPECT_EQ(tefe.EmbeddingDims(1), 24); in TEST_F()132 EXPECT_EQ(tefe.EmbeddingDims(0), 8); in TEST_F()138 EXPECT_EQ(tefe.EmbeddingDims(1), 16); in TEST_F()
72 int EmbeddingDims(int index) const { return embedding_dims_[index]; } in EmbeddingDims() function