Searched defs:grad_acc (Results 1 – 3 of 3) sorted by relevance
108 auto grad_acc = (AccumulateGrad*)self->cdata.get(); in accumulateGradVar() local
333 auto& grad_acc = dynamic_cast<AccumulateGrad&>(*grad_acc_fn); in _process_backward_mode_ad() local
492 if (auto grad_acc = in THPVariable_clear() local