Searched defs:MaxPoolGradWithArgmaxV2 (Results 1 – 6 of 6) sorted by relevance
37 …MaxPoolGradWithArgmaxV2() : BaseOperator(kNameMaxPoolGradWithArgmaxV2) { InitIOName({"x", "grad", … in MaxPoolGradWithArgmaxV2() function
30 class MaxPoolGradWithArgmaxV2(nn.Cell): class
23 __global__ void MaxPoolGradWithArgmaxV2(const T *dy, const S *index, const int64_t x_hw, const int6… in MaxPoolGradWithArgmaxV2() function
134 "MaxPoolGradWithArgmaxV2": { object
1089 class MaxPoolGradWithArgmaxV2(Primitive): class