Searched refs:embedding_dims_ (Results 1 – 2 of 2) sorted by relevance
55 embedding_dims_.push_back(parsed_dim); in Init()58 (embedding_fml_.size() != embedding_dims_.size())) { in Init()61 << "; #dims = " << embedding_dims_.size(); in Init()
62 int NumEmbeddings() const { return embedding_dims_.size(); } in NumEmbeddings()72 int EmbeddingDims(int index) const { return embedding_dims_[index]; } in EmbeddingDims()75 const std::vector<int> &embedding_dims() const { return embedding_dims_; } in embedding_dims()108 std::vector<int> embedding_dims_; variable