Searched defs:flat_grads (Results 1 – 3 of 3) sorted by relevance
171 def replace_flat_tensors_for_gradients(xs, flat_grads): argument
171 def backward(ctx, *flat_grads):
373 def backward(ctx, *flat_grads):