Searched refs:reference_tensors (Results 1 – 1 of 1) sorted by relevance
527 def checkTrace(self, func, reference_tensors, input_tensors=None, argument542 input_tensors = reference_tensors555 nograd_inputs = reference_tensors557 recording_inputs = do_input_map(lambda t: t.clone().requires_grad_(), reference_tensors)560 recording_inputs = reference_tensors604 … recording_inputs = do_input_map(lambda t: Variable(t, requires_grad=True), reference_tensors)