lhutton1 opened a new pull request #6248: URL: https://github.com/apache/incubator-tvm/pull/6248
This patch adds support for various pooling operators, namely: average pool 2d, global max pool 2d, global average pool 2d and l2 pooling (an equivalent combination of relay operators). For fp32: - nn.avg_pool2d - nn.global_max_pool2d - nn.global_avg_pool2d - power(2) + nn.avg_pool2d + sqrt -> L2 pooling (only supports fp32) For uint8: - cast + nn.avg_pool2d + cast - nn.global_max_pool2d - cast + nn.global_avg_pool2d + cast Tests updated to reflect these changes. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
