Searched defs:MaxPoolWithArgmax (Results 1 – 2 of 2) sorted by relevance
22 __global__ void MaxPoolWithArgmax(const T* input, in MaxPoolWithArgmax() function
1758 class MaxPoolWithArgmax(_Pool): class