Searched defs:values_and_indices (Results 1 – 2 of 2) sorted by relevance
179 XlaBuilder* builder) -> StatusOr<std::vector<XlaOp>> { in TopKWithPartitions()
1247 def values_and_indices(t): function