Searched refs:replace_flat_tensors_for_gradients (Results 1 – 3 of 3) sorted by relevance
171 def replace_flat_tensors_for_gradients(xs, flat_grads): function
1126 flat_grad = composite_tensor_gradient.replace_flat_tensors_for_gradients(
502 return composite_tensor_gradient.replace_flat_tensors_for_gradients(