Searched refs:embedding_dims_ (Results 1 – 2 of 2) sorted by relevance
66 int NumEmbeddings() const { return embedding_dims_.size(); } in NumEmbeddings()92 std::vector<int> embedding_dims_; variable
63 embedding_dims_.push_back(dim); in Setup()