xla

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

#8038

Merged