Searched refs:capture_autograd_function (Results 1 – 2 of 2) sorted by relevance
425 capture_autograd_function = True variable
661 and config.capture_autograd_function