Searched refs:testInTopK (Results 1 – 2 of 2) sorted by relevance
352 def testInTopK(self, dtype): member in XlaSortOpTest
1389 def testInTopK(self): member in TestUpgrade