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
..
2019-12-20 22:35:39 +03:00
2019-12-20 22:35:39 +03:00
2019-12-20 22:35:39 +03:00
2019-12-20 22:35:39 +03:00
2020-01-20 12:59:12 +03:00
2020-01-04 13:27:50 +03:00
2020-01-04 13:27:50 +03:00
2019-12-20 22:35:39 +03:00
2019-06-06 15:21:15 +03:00
2019-12-20 22:35:39 +03:00
2019-12-20 22:35:39 +03:00
2019-12-20 22:35:39 +03:00
2019-08-26 19:57:51 +03:00
2020-01-28 18:23:07 +03:00
2020-01-22 13:59:36 +03:00
2020-01-06 22:35:05 +03:00
2019-12-20 22:35:39 +03:00
2020-01-04 13:27:50 +03:00
2019-12-20 22:35:39 +03:00
2019-12-20 22:35:39 +03:00
2019-12-21 08:07:13 +03:00
2019-08-05 11:27:05 +10:00
2020-01-04 13:27:50 +03:00
2019-06-06 15:21:15 +03:00

Generic ops

Put any cross platform operations here. Anything platform specific should be put in: in ops/helpers.