Searched defs:embedding_lookup (Results 1 – 2 of 2) sorted by relevance
35 auto embedding_lookup = std::make_shared<EmbeddingLookup>(); in TEST_F() local
694 auto embedding_lookup = CreateEmbeddingLookup(graph, ele->input(1), indices); in CacheEmbeddingForEval() local