Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
cavis/libnd4j/include/ops/declarable/generic/nn
History
Alex Black 18c01f5bdc
Add SameDiff memory reuse memory manager (array cache) (#39) 
* Attention op comments

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

* ArrayCacheMemoryMgr - first pass

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

* Tweak array cache for use with SameDiff identity arrays

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

* ArrayCacheMemoryMgr javadoc and properly get max memory

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

* LRU cache policy + add tests

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

* Fixes

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

* Resize arrays internally if required for ArrayCacheMemoryMgr

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

* Test improvement

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

* Small polish

Signed-off-by: AlexDBlack <blacka101@gmail.com>
 		2019-11-12 21:15:44 +11:00
..
convo Shyrma concat (#24) 2019-11-03 12:37:19 +02: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 bnorm bp (#41) 2019-11-12 11:58:48 +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 Add SameDiff memory reuse memory manager (array cache) (#39) 2019-11-12 21:15:44 +11: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