Searched refs:need_grad (Results 1 – 1 of 1) sorted by relevance
207 void set_grad(const bool &need_grad) { need_grad_ = need_grad; } in set_grad() argument