Searched refs:grads_ (Results 1 – 2 of 2) sorted by relevance
168 std::unordered_map<Key, WeightPtr> grads_; variable
160 if (grads_.count(key) == 0) { in InitGrad()161 grads_[key] = grad; in InitGrad()