Searched refs:is_embedding_ (Results 1 – 2 of 2) sorted by relevance
150 if ((weights_.count(key) == 0) || (is_embedding_[key] && weights_.count(key) != 0)) { in InitWeight()154 is_embedding_[key] = false; in InitWeight()259 is_embedding_[key] = true; in InitEmbeddingTable()265 bool ParameterServer::HasWeight(const Key &key) { return (weights_.count(key) > 0 && !is_embedding_… in HasWeight()314 if (!is_embedding_[key]) { in UpdateWeights()338 … optim_inputs_shape_[key], worker_num_, is_embedding_[key]); in AccumGrad()
167 std::unordered_map<Key, bool> is_embedding_; variable