- 20 Aug, 2018 1 commit
-
-
* Improve x86 pooling and concat * Fix * Fix test concatenate correct layout * Add conditional vectorize * Fix lint * Modify schedule for global pooling * Fix * Fix warning * Fix alter layout test * Remove vectorization for pooling when using 4D layout * Remove vectorization for 4D concat * Fix concatenate layout * Fix concatenate schedule * Fix concat * Fix lint * Fix concat * Simplify pooling logic * Update docstring * Fix test topi pooling * Small changes
Yao Wang committed
-
- 09 Jul, 2018 1 commit
-
-
Yizhi Liu committed
-
- 28 Jun, 2018 1 commit
-
-
Tianqi Chen committed
-
- 29 May, 2018 17 commits
-
-
Siva committed
-
Yizhi Liu committed
-
Pariksheet Pinjari committed
-
Yan Huang committed
-
* Add conv2d max_pool backward op * Added tests * Fix testing * Address comments * Change dot to matmul * Address comments * Break down indicator function * Make greater, less numpy compatible
yuruofeifei committed -
* make shape inference of BatchNorm layout neutral * refactor to use the axis variable to do BatchNorm shape inference * refactor to use the axis variable to do BatchNorm shape inference * add unittest to the axis param for batch norm shape inference
guopinglong committed -
* Add creating gradient symbol * Fix lint * Address comments * Fix typo * Address comment
Yao Wang committed -
Tianqi Chen committed
-
Tianqi Chen committed
-
Tianqi Chen committed
-
Tianqi Chen committed
-
* register depthconv, elemwise * use global elemwise schedule for relu
Yuwei HU committed -
Yuwei HU committed
-
Tianqi Chen committed
-
* [TOP] level4 except argmax/min, correct split * [DOCS] Add doc generator for top
Tianqi Chen committed -
* [TOP] complete level2 * [TOP] add split
Tianqi Chen committed -
Tianqi Chen committed
-