Searched defs:fn_cls (Results 1 – 1 of 1) sorted by relevance
624 def produce_trampoline_autograd_apply(fn_cls): argument640 def __init__(self, fn_cls, **kwargs) -> None: argument