Searched defs:set_grad (Results 1 – 3 of 3) sorted by relevance
215 def set_grad(self, flag=True): member in SequentialCell389 def set_grad(self, flag=True): member in CellList
207 void set_grad(const bool &need_grad) { need_grad_ = need_grad; } in set_grad() function
1215 def set_grad(self, requires_grad=True): member in Cell