Searched refs:embedding_num_features_size (Results 1 – 3 of 3) sorted by relevance
168 int embedding_num_features_size() const override { in embedding_num_features_size() function173 if (!InRangeIndex(i, embedding_num_features_size(), in embedding_num_features()
292 virtual int embedding_num_features_size() const = 0;
291 for (int i = 0; i < model_->embedding_num_features_size(); ++i) { in EmbeddingNetwork()