xla

b79a46b0 - Implement torch diagflat, and movedim as well as it is also needed as a helper function (#7378) (#8038)