Searched refs:embedding_num_features_size (Results 1 – 4 of 4) sorted by relevance
66 if (embeddings_size() != embedding_num_features_size()) { in GetNumEmbeddingSpaces()68 << " != " << embedding_num_features_size(); in GetNumEmbeddingSpaces()71 std::min(embeddings_size(), embedding_num_features_size())); in GetNumEmbeddingSpaces()94 TC_DCHECK(InRange(i, embedding_num_features_size())); in GetNumFeaturesInEmbeddingSpace()306 virtual int embedding_num_features_size() const = 0;
218 int embedding_num_features_size() const override { in embedding_num_features_size() function219 return proto_->embedding_num_features_size(); in embedding_num_features_size()223 TC_DCHECK(InRange(i, embedding_num_features_size())); in embedding_num_features()
36 int embedding_num_features_size() const override { in embedding_num_features_size() function94 int embedding_num_features_size() const override { in embedding_num_features_size() function95 return embedding_params_->embedding_num_features_size(); in embedding_num_features_size()
196 int embedding_num_features_size() const override { in embedding_num_features_size() function197 return trimmed_proto_.embedding_num_features_size(); in embedding_num_features_size()201 TC_DCHECK(InRange(i, embedding_num_features_size())); in embedding_num_features()