Searched refs:argmax_v2 (Results 1 – 1 of 1) sorted by relevance
137 return argmax_v2(input, axis, output_type, name)141 def argmax_v2(input, function