guberti opened a new pull request, #13051: URL: https://github.com/apache/tvm/pull/13051
This pull request contains two changes: 1. Adds support for specifying an `out_layout` to the `Tensordot`-based convolution schedules for Arm Cortex-M. This is useful, as when compiling models like MobileNetV1, we will want to alternate between NHWC and NCHW layers (see [this comment](https://github.com/apache/tvm/pull/12969#issuecomment-1268113822) for more). 2. Re-organizes the Corstone300 tests for regular, grouped, and depthwise conv2d to use a shared base class. This removes a lot of code re-use that used to exist in `tests/python/relay/strategy/arm_cpu`. I chose to make these changes as one PR instead of two, since I had to change some of the testing logic to support the `out_layout` parameter (and copying that into four places would be kinda gross). Would love a look from @mkatanbaf! -- 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. To unsubscribe, e-mail: commits-unsubscr...@tvm.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
