Searched refs:embedding_num_features (Results 1 – 4 of 4) sorted by relevance
295 virtual int embedding_num_features(int i) const = 0;
294 offset_sum += matrix.cols * model_->embedding_num_features(i); in EmbeddingNetwork()
172 int embedding_num_features(int i) const override { in embedding_num_features() function
216 input_size += embedding_num_features(i) * embeddings.cols; in ValidityChecking()