tree cae5e096c4ee2188a2d2a7ad558bfc3e338ffe3d
parent 092fe1b3fa7660d380a7b3816acf47e7d0b7b496
author A. Unique TensorFlower <gardener@tensorflow.org> 1657651732 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1657652093 -0700

Track op names in dtensor emitted ops

This change allows the original op name to show up in xprof of DTensor
emitted reduce ops.

PiperOrigin-RevId: 460516949