Searched defs:maxpool (Results 1 – 2 of 2) sorted by relevance
47 CNodePtr CreateMaxPoolWithArgmax(const FuncGraphPtr &graph, const CNodePtr &maxpool) { in CreateMaxPoolWithArgmax()89 void SetNodeAttrs(const CNodePtr &maxpool, const CNodePtr &maxpool_grad, const CNodePtr &maxpool_ar… in SetNodeAttrs()122 VectorRef maxpool({prim::kPrimMaxPool, X}); in DefinePattern() local134 auto maxpool = GetMaxPool(maxpool_grad); in Process() local
457 auto maxpool = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local528 auto maxpool = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local605 auto maxpool = static_cast<PoolingParameter *>(malloc(sizeof(PoolingParameter))); in TEST_F() local