Searched refs:EmbeddingLookupOpTest (Results 1 – 2 of 2) sorted by relevance
87 TEST(EmbeddingLookupOpTest, SimpleTest) { in TEST() argument102 TEST(EmbeddingLookupOpTest, SimpleTestMean) { in TEST() argument118 TEST(EmbeddingLookupOpTest, SimpleTestSqrtn) { in TEST() argument138 TEST(EmbeddingLookupOpTest, Indices3DTest) { in TEST() argument
96 TEST(EmbeddingLookupOpTest, SimpleTest) { in TEST() argument