cavis/libnd4j/include/ops/declarable/platform/mkldnn
Yurii Shyrma cae5ef4180 Shyrma depthconv (#156)
* - implementation of depthwise_conv2d (both ff/bp) based on mkl dnn api

* - minor corrections in deconv3d

Signed-off-by: Yurii <iuriish@yahoo.com>

* - remove unnecessary time test

Signed-off-by: Yurii <iuriish@yahoo.com>

* - update mkl dnn version in cmake

Signed-off-by: Yurii <iuriish@yahoo.com>

* - take into account several notes given by pr reviewer

Signed-off-by: Yurii <iuriish@yahoo.com>

* - fix bug in depthwise conv2d op based on mkl

Signed-off-by: Yurii <iuriish@yahoo.com>
2020-01-11 07:36:40 +03:00
..
avgpooling2d.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
avgpooling2d_bp.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
avgpooling3d.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
avgpooling3d_bp.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
batchnorm.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
conv2d.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
conv3d.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
deconv2d.cpp Shyrma depthconv (#156) 2020-01-11 07:36:40 +03:00
deconv2d_tf.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
deconv3d.cpp Shyrma depthconv (#156) 2020-01-11 07:36:40 +03:00
depthwiseConv2d.cpp Shyrma depthconv (#156) 2020-01-11 07:36:40 +03:00
lrn.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
lstmLayer.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
maxpooling2d.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
maxpooling2d_bp.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
maxpooling3d.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
maxpooling_3d_bp.cpp allow MKL-DNN on non-AVX machines (#104) 2019-12-04 09:11:37 +03:00
mkldnnUtils.cpp DNNL/MKLDNN dilated causal conv1d + betainc (#103) 2019-12-04 14:50:17 +03:00
mkldnnUtils.h Shyrma depthconv (#156) 2020-01-11 07:36:40 +03:00