tree e2f3a42046bbe33e46f122955a413b2747cec5c4
parent 4f0759177bc77b3b8247622e8a727539bc76d814
author A. Unique TensorFlower <gardener@tensorflow.org> 1609943806 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1609944121 -0800

Cleans up a newly added XlaVariadicSort op to take Input instead of Attr.

`dimension` and `is_stable` were passed as attributes before but are now
allowed to be inputs. However, they are constrained to be compile-time
constants for now.

Also add proper kernel registrations for CPU and GP (which was missing in the
previous change).

PiperOrigin-RevId: 350341893
Change-Id: Iab0554ed9ba12343f388f9963e4624c9823a227e