cavis/libnd4j/include/ops/declarable/generic/nn
Alex Black 2052ce7026
Various pre-release fixes (#111)
* Various fixes

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* Fix default dtypes for MaxPoolWithArgmax

Signed-off-by: AlexDBlack <blacka101@gmail.com>
2019-12-05 14:20:03 +11:00
..
convo Shyrma casual conv1d (#90) 2019-11-29 14:14:30 +03:00
pooling Various pre-release fixes (#111) 2019-12-05 14:20:03 +11:00
apply_sgd.cpp [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
batchnorm.cpp [WIP] Weekly update of repo (#8390) 2019-11-13 17:15:18 +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 [WIP] Weekly update of repo (#8390) 2019-11-13 17:15:18 +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