[XLA] add SortKeyVal to the local Python client.
This operation corresponds to the version of the Sort HLO with three arguments, but we give it a separate name (SortKeyVal instead of Sort) for compatibility with SWIG. PiperOrigin-RevId: 209427551
Loading
Please sign in to comment