cavis/libnd4j/include
Yurii Shyrma 7a7ee4b021 Shyrma cudnn (#192)
* - implementation of cudnn batchnorm_bp op

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

* - testing and fixing bugs in batchnorm_bp based on cudnn api

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

* - move pooling mkl code and delete some unnecessary files

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

* - implementation and testing cudnn pooling2d ops (avg/max, ff/bp)

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

* - implementation and testing cudnn pooling 3d (ff/bp) ops

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

* - provide ff step in case of cudnn maxpool3d_bp op

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

* - remove half type from set of supported types in mkl dpethwise conv op

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

* - bring back cudaStreamSynchronize in batchnorm and pooling cudnn ops

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

Co-authored-by: raver119 <raver119@gmail.com>
2020-01-28 18:23:07 +03:00
..
array missing countOut for GPU 2020-01-24 15:37:24 +03:00
cnpy [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
exceptions [WIP] Memory limits (#167) 2020-01-24 10:11:09 +03:00
execution execution mode (#183) 2020-01-27 10:00:07 +03:00
graph execution mode (#183) 2020-01-27 10:00:07 +03:00
helpers cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
indexing Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
loops cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
memory [WIP] Memory limits (#167) 2020-01-24 10:11:09 +03:00
ops Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00
performance/benchmarking Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
types Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
Status.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
buffer.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
cblas.h Platform helpers (#8216) 2019-09-11 21:50:28 +03:00
cblas_enum_conversion.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
config.h.in cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
dll.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
enum_boilerplate.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
msvc.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
nd4jmalloc.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
nd4jmemset.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
op_boilerplate.h String changes (#3) 2020-01-04 13:27:50 +03:00
op_enums.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
openmp_pragmas.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
optype.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
pairwise_util.h Shugeo random uniform int (#30) 2019-11-06 12:49:27 +02:00
platform_boilerplate.h cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
platformmath.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
play.h cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
pointercast.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
templatemath.h Shugeo atomic tests (#97) 2019-12-02 21:40:54 +03:00
type_boilerplate.h String changes (#3) 2020-01-04 13:27:50 +03:00
util.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00