- 25 Apr, 2018 1 commit
-
-
* support dilation in conv2d and depthwise_conv2d * handle dilated conv in extern libs (cudnn, miopen)
Yuwei Hu committed
-
- 02 Mar, 2018 1 commit
-
-
* explicit import testing * Enable init api for extension modules
Tianqi Chen committed
-
- 16 Jan, 2018 1 commit
-
-
* add basic x86 schedules * parallelize & vectorize batchnorm + relu * fuse conv into bn + relu * move rc loop to outer * add nhwc conv * change weight layout to hwcf * conv + bn + relu fusion for nhwc conv * fix conv_nhwc schedule when no fusion * clean up default parallel schedules * simplify elemwise parallel * fix elemwise parallel for batch == 1 * update nhwc conv test * fix and add comment * fix lint * remove redundant import * remove default multithreading for some ops * remove default multithreading for global pool
masahi committed
-