cavis/libnd4j/include/helpers/cuda
Yurii Shyrma 029a69a835
Shyrma bn mkl bp (#14)
* - write code for new batchnorm backprop

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

* - testing batchnorm backprop

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

* - write code for batchnorm backprop based on mkl dnn api

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

* - testing and fixing bugs in batchnorm_bp mkl dnn

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

* - made corrections required by reviewer

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

* - change name in java wrapper for batchnorm op

Signed-off-by: Yurii <iuriish@yahoo.com>
2019-10-26 14:14:21 +03:00
..
ConstantHelper.cu Snapshot update (#8194) 2019-09-03 22:02:02 +03:00
ConstantShapeHelper.cu Shyrma bn mkl bp (#14) 2019-10-26 14:14:21 +03:00
ConstantTadHelper.cu Snapshot update (#8194) 2019-09-03 22:02:02 +03:00
PointersManager.cu Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
TrueBroadcastHelper.cu [WIP] Broadcast changes (#8257) 2019-10-01 09:10:19 +03:00