xla
8301c055 - [Functionalization] Lower diagonal_scatter properly

Commit
2 years ago
[Functionalization] Lower diagonal_scatter properly Summary: This change replaces the old implementation with DiagonalViewUpdate. Test Plan: PJRT_DEVICE=CPU python ../test/test_view_ops.py -v -k TestViewOpsXLA.test_diagonal
Author
Parents
Loading