Searched refs:kPrimMaxPoolGrad (Results 1 – 4 of 4) sorted by relevance
46 {prim::kPrimMaxPoolGrad->name(), {{0, 1, 2}, {0}}},
123 VectorRef pattern({prim::kPrimMaxPoolGrad, X, maxpool, Y}); in DefinePattern()
367 if (apply_function_name == prim::kPrimMaxPoolGrad->name() && in TbeClearZeroNodeFusion()438 } else if (apply_function_name == prim::kPrimMaxPoolGrad->name() && in KernelBuildPreprocess()
285 inline const PrimitivePtr kPrimMaxPoolGrad = std::make_shared<Primitive>("MaxPoolGrad"); variable