Skip to content
Snippets Groups Projects
Commit 9ac46bea authored by Leyuan Wang's avatar Leyuan Wang Committed by Tianqi Chen
Browse files

[TOPI] Fix conv2d for small input channels (#331)

* __init__ updated

* pull request updated

* build_module added

* typo fixed

* another typo fixed

* conv2d gpu scheduler for two layouts moved to tvm

* changes made according to CR

* conv2d_nchw formating updated, conv2d_hwcn tests updated

* lint error fixed

* element wise operator schedule fusing fixed for conv2d

* conv2d_nchw topi test added, all resnet workloads now pass

* conv compute lint error fixed

* fixed python 3 compatibility problem

* conv2d tensor input support added, test typo fixed, ir_pass.Simplify changed to util.get_const_int

* fixed channel numer < 4 error, also made sure other splitting factor woudn't be 0
parent 0ad590c0
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment