Searched defs:pooling_prim (Results 1 – 4 of 4) sorted by relevance
22 auto pooling_prim = primitive->value_as_MaxPoolFusion(); in IsSupport() local30 int MaxPoolingNPUOp::SetPoolingParam(const schema::MaxPoolFusion *pooling_prim) { in SetPoolingParam()74 auto pooling_prim = primitive->value_as_MaxPoolFusion(); in Init() local
22 auto pooling_prim = primitive->value_as_AvgPoolFusion(); in IsSupport() local38 int AvgPoolingNPUOp::SetPoolingParam(const schema::AvgPoolFusion *pooling_prim) { in SetPoolingParam()82 auto pooling_prim = primitive->value_as_AvgPoolFusion(); in Init() local
64 auto pooling_prim = static_cast<const schema::Primitive *>(primitive); in PopulateAvgPoolParameter() local114 auto pooling_prim = static_cast<const schema::Primitive *>(primitive); in PopulateMaxPoolParameter() local
27 auto pooling_prim = primitive->value_as_Pooling(); in PopulatePoolingParameter() local