Searched refs:grad_to_source (Results 1 – 1 of 1) sorted by relevance
50 grad_to_source=None, argument56 self.grad_to_source = grad_to_source or {}197 self.grad_to_source = {}270 self.grad_to_source[p.grad] = grad_source284 and v not in self.grad_to_source304 elif tensor_value in self.grad_to_source:305 builder = VariableBuilder(tx, self.grad_to_source[tensor_value])