David Hou 3151d91f6e 3x3 winograd convs (#1675) ...

* winograd

* simplify local groups code

* comment

* respects self.opts.has_local

* always simplify ones

* make mypy happy

* move reshape, WINO flag

* wino flag, simple forward backward test for wino

* extra wino test

* merge oops

* comments

* axis_needs_valid -> axis_is_masked

* don't delete needs_valid (it's unused though)

* make linter happy

* make linter happy

* smaller test

* change number

* make wino tests very small

2023-09-03 07:29:43 -07:00

4.3 KiB

Raw Blame History

View Raw