Searched defs:pooling_prim (Results 1 – 3 of 3) sorted by relevance
22 auto pooling_prim = primitive->value_as_MaxPoolFusion(); in IsSupport() local30 int MaxPoolingNPUOp::SetPoolingParam(const schema::MaxPoolFusion *pooling_prim) { in SetPoolingParam()81 auto pooling_prim = primitive->value_as_MaxPoolFusion(); in Init() local
26 auto pooling_prim = primitive->value_as_AvgPoolFusion(); in IsSupport() local55 int AvgPoolingNPUOp::SetPoolingParam(const schema::AvgPoolFusion *pooling_prim) { in SetPoolingParam()105 auto pooling_prim = primitive->value_as_AvgPoolFusion(); in Init() local
73 auto pooling_prim = static_cast<const schema::Primitive *>(primitive); in PopulateAvgPoolParameter() local129 auto pooling_prim = static_cast<const schema::Primitive *>(primitive); in PopulateMaxPoolParameter() local