Searched defs:MaxPoolWithArgmax (Results 1 – 4 of 4) sorted by relevance
35 …MaxPoolWithArgmax() : BaseOperator(kNameMaxPoolWithArgmax) { InitIOName({"x"}, {"output", "mask"})… in MaxPoolWithArgmax() function
21 __global__ void MaxPoolWithArgmax(const T *input, const int n, const int c, const int h, const int … in MaxPoolWithArgmax() function
70 class MaxPoolWithArgmax(Primitive): class
64 "MaxPoolWithArgmax": { object