Searched refs:embeddings_size (Results 1 – 3 of 3) sorted by relevance
101 CheckIndex(i, embeddings_size(), "embedding matrix"); in GetEmbeddingMatrix()199 virtual int embeddings_size() const = 0;
60 int embeddings_size() const override { return SafeGetNumInputChunks(); } in embeddings_size() function
210 for (int i = 0; i < embeddings_size(); ++i) { in ValidityChecking()