tree 2c8ffbbc603762045b03a3331915275893041dcd
parent 10c84d967a353866e776cf8498460bac23586421
author Michael Delorimier <mdel@google.com> 1639517192 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1639517541 -0800

Switches MatrixDiagOp to MlirXlaOpKernel for lowering from TF to HLO in the tf2xla bridge.

PiperOrigin-RevId: 416379755
Change-Id: I3ff5713562c6e3d39a10d14d4259245dd2617e2a