cavis

History

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
..
cudnn	Shyrma cudnn (#192 )	2020-01-28 18:23:07 +03:00
mkldnn	Shyrma cudnn (#192 )	2020-01-28 18:23:07 +03:00
README.md	Platform helpers (#8216 )	2019-09-11 21:50:28 +03:00

This folder contains platform-specific optimized implementations for custom operations