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 |
||
---|---|---|
.. | ||
BUILD | ||
xla_op_registry.clif | ||
xla.py |