cavis/libnd4j/include/ops
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
..
declarable Shyrma depthconv (#156) 2020-01-11 07:36:40 +03:00
impl Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
BroadcastBoolOpsTuple.h [WIP] CUDA tests (#95) 2019-12-02 21:37:21 +03:00
BroadcastIntOpsTuple.h [WIP] CUDA tests (#95) 2019-12-02 21:37:21 +03:00
BroadcastOpsTuple.h [WIP] CUDA tests (#95) 2019-12-02 21:37:21 +03:00
InputType.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
gemm.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
meta_ops.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
ops.h Shyrma adjust (#98) 2019-12-03 09:40:45 +03:00
random_ops.h Gamma and Poisson distributions (#27) 2019-11-04 15:42:28 +02:00
special_random_ops.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
specials.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
specials_cuda.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
specials_sparse.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00