Searched defs:use_tape_cache (Results 1 – 1 of 1) sorted by relevance
74 def __init__(self, func, Tout, is_grad_func, use_tape_cache=True): argument286 use_tape_cache=True): argument379 def _eager_py_func(func, inp, Tout, name=None, use_tape_cache=True): argument