Searched refs:grad_fun (Results 1 – 2 of 2) sorted by relevance
864 grad_fun = (868 return grad_fun(883 dx, dscale, doffset, _, _ = grad_fun(
858 def grad_fun(dresult): function860 tape.record_operation("_copy", [new_tensor], [self], grad_fun)