Searched refs:SortKeyVal (Results 1 – 4 of 4) sorted by relevance
356 LocalOp SortKeyVal(const LocalOp& keys, const LocalOp& values,
372 %unignore xla::swig::ComputationBuilder::SortKeyVal;
1737 def SortKeyVal(self, keys, values, dimension=-1): member in ComputationBuilder1739 return self._client.SortKeyVal(keys, values, dimension)
712 LocalOp ComputationBuilder::SortKeyVal(const LocalOp& keys, in SortKeyVal() function in xla::swig::ComputationBuilder