Searched refs:is_grad_func (Results 1 – 1 of 1) sorted by relevance
53 def __init__(self, func, Tout, is_grad_func): argument65 self._is_grad_func = is_grad_func241 is_grad_func=False, argument252 func = EagerFunc(func, Tout, is_grad_func)306 is_grad_func=True)