Searched refs:_replace_tensors_for_gradient (Results 1 – 1 of 1) sorted by relevance
129 def _replace_tensors_for_gradient(x, grad): function150 _replace_tensors_for_gradient,183 return [_replace_tensors_for_gradient(x, grad) for x, grad in zip(xs, grads)]