Files
apache--tvm/python
Xiyou Zhou 46a8498ba9 [MetaSchedule] Enhance Conv2d NCHW Winograd Schedule Rules (#12127)
* Update winograd schedule rules.

* Remove extra part for setting local storage scope.

* Fix bgemm schedule.

* Add winograd tile size to annotation.

* Finish winograd schedule rules.

* Process add relu.

* Modify to nchw rules.

* Add missing nchw output rules.

* Add winograd conv2d nchw search space test.

* Fix lint.

* Leave consumer of output to autoinline.

* Remove bgemm rules.

* Remove bgemm schedule rule annotation.

* Update unit test.

* Fix test case.
2022-08-03 14:42:08 -07:00
..