cavis/libnd4j/include/ops/declarable/platform/cudnn
Yurii Shyrma 948646b32d
Shyrma mkl test (#211)
* - provide nhwc format in mkl conv ops

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

* - corrections in mkl conv3d

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

* - corrections in mkl batchnorm

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

* - corrections in mkl maxpooling2d

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

* - add format format_tag::any to outputs in mkl conv ops

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

* - complete corrections in mkl conv ops

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

* - add test for comparison of execution speeds of mkl conv2d op with different weights format

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

* - take into account order f in mkl conv ops

Signed-off-by: Yurii <iuriish@yahoo.com>
2020-02-06 21:12:54 +03:00
..
avgpool2d.cu Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00
avgpool3d.cu Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00
batchnorm.cu Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00
conv2d.cu cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
conv3d.cu cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
cudnnUtils.cu Shyrma mkl test (#211) 2020-02-06 21:12:54 +03:00
cudnnUtils.h Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00
depthwiseConv2d.cu cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
maxpool2d.cu Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00
maxpool3d.cu Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00