A. Unique TensorFlower 682d130cf6 Exposes a variadic form of XlaSort. (Previously, we had XlaSort for 1 argument, ...

and XlaKeyValueSort for 2 arguments.)

The new op, XlaVariadicSort, fully exposes the xla::Sort functionality: can
take an arbitrary list of operands, can take dimension and
is_stable arguments, and a comparator function.

PiperOrigin-RevId: 348581487
Change-Id: I82b4ff0c37d433541167584475b004626bc8c82c

2020-12-21 22:35:25 -08:00

..

BUILD

…

xla_op_registry.clif

…

xla.py

…