Searched defs:AvgPool (Results 1 – 3 of 3) sorted by relevance
35 AvgPool() : PrimitiveC(kNameAvgPool) { InitIOName({"x"}, {"output"}); } in AvgPool() function36 explicit AvgPool(const std::string k_name) : PrimitiveC(k_name) { InitIOName({"x"}, {"output"}); } in AvgPool() function
340 MS_TRT_CONVERTER_FUNC_REG(AvgPool) { return AddPoolingLayer(node, context, nvinfer1::PoolingType::k… in MS_TRT_CONVERTER_FUNC_REG() argument
1989 class AvgPool(_Pool): class