Lines Matching refs:embedding_params_
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()
143 std::shared_ptr<EmbeddingParams> embedding_params_; variable