[TOPI] Add out_dtype argument for conv2d; Add x86 schedules (#646)
* [TOPI] Add out_dtype argument for conv2d; Add x86 schedules * Fix * Fix lint * Fix
Showing
- topi/python/topi/__init__.py 1 addition, 0 deletionstopi/python/topi/__init__.py
- topi/python/topi/nn/conv2d.py 41 additions, 39 deletionstopi/python/topi/nn/conv2d.py
- topi/python/topi/nn/depthwise_conv2d.py 3 additions, 3 deletionstopi/python/topi/nn/depthwise_conv2d.py
- topi/python/topi/rasp/conv2d.py 10 additions, 9 deletionstopi/python/topi/rasp/conv2d.py
- topi/python/topi/rasp/depthwise_conv2d.py 14 additions, 14 deletionstopi/python/topi/rasp/depthwise_conv2d.py
- topi/python/topi/x86/__init__.py 5 additions, 0 deletionstopi/python/topi/x86/__init__.py
- topi/python/topi/x86/conv2d.py 37 additions, 0 deletionstopi/python/topi/x86/conv2d.py
Loading
Please register or sign in to comment