Searched defs:EmbeddingLookup (Results 1 – 8 of 8) sorted by relevance
35 …EmbeddingLookup() : PrimitiveC(kNameEmbeddingLookup) { InitIOName({"params", "indices", "offset"},… in EmbeddingLookup() function
50 int EmbeddingLookup(float *input_data, const int *ids, float *output_data, const EmbeddingLookupPar… in EmbeddingLookup() function
45 class EmbeddingLookup(Cell): class
156 class EmbeddingLookup(Cell): class
103 class EmbeddingLookup(nn.Cell): class
127 class EmbeddingLookup(nn.Cell): class
350 class EmbeddingLookup(nn.Cell): class
5830 class EmbeddingLookup(PrimitiveWithCheck): class