cavis/libnd4j/include/ops/declarable/generic/nn
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
..
convo SameDiff cleanup and fixes (#12) 2019-10-26 12:38:08 +11:00
pooling Platform helpers (#8216) 2019-09-11 21:50:28 +03:00
apply_sgd.cpp [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
batchnorm.cpp Shyrma bn mkl bp (#14) 2019-10-26 14:14:21 +03:00
dot_product_attention.cpp Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
fusedBatchNorm.cpp Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
logSoftmax.cpp Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
lrn.cpp Platform helpers (#8216) 2019-09-11 21:50:28 +03:00
multi_head_dot_product_attention.cpp Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
relu_layer.cpp Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
softmax.cpp [WIP] Error handling (#169) 2019-08-26 19:57:51 +03:00