Searched refs:embedding_params_ (Results 1 – 1 of 1) sorted by relevance
86 return embedding_params_; in GetEmbeddingParams()91 return embedding_params_->embeddings_size(); in embeddings_size()95 return embedding_params_->embedding_num_features_size(); in embedding_num_features_size()99 return embedding_params_->embedding_num_features(i); in embedding_num_features()103 return embedding_params_->embeddings_num_rows(i); in embeddings_num_rows()107 return embedding_params_->embeddings_num_cols(i); in embeddings_num_cols()111 return embedding_params_->embeddings_weights(i); in embeddings_weights()115 return embedding_params_->embeddings_quant_type(i); in embeddings_quant_type()119 return embedding_params_->embeddings_quant_scales(i); in embeddings_quant_scales()137 embedding_params_(std::move(embedding_params)) {} in ModelParams()[all …]