Remove calculation of unnecessary matrix columns in SVD gradient (#15801)
* remove calculation of unnecessary matrix columns in svd gradient * simplify expression for svd gradient when compute_uv=False * assign Operation attribute to local variable
Loading
Please sign in to comment