cavis/libnd4j/include/ops/declarable/generic/nn
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
..
convo Shyrma depthconv (#156) 2020-01-11 07:36:40 +03:00
pooling Update master (#8511) 2019-12-06 11:10:44 +03:00
apply_sgd.cpp [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
batchnorm.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
dot_product_attention.cpp Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
fusedBatchNorm.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
logSoftmax.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
lrn.cpp Platform helpers (#8216) 2019-09-11 21:50:28 +03:00
multi_head_dot_product_attention.cpp [WIP] Weekly update of repo (#8390) 2019-11-13 17:15:18 +03:00
relu_layer.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
softmax.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00