cavis/libnd4j/include/ops
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
..
declarable Add SameDiff memory reuse memory manager (array cache) (#39) 2019-11-12 21:15:44 +11:00
impl Added broadcastable IGamma and IGammac ops. 2019-10-16 13:58:32 +03:00
BroadcastBoolOpsTuple.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
BroadcastIntOpsTuple.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
BroadcastOpsTuple.h Added broadcastable IGamma and IGammac ops. 2019-10-16 13:58:32 +03:00
InputType.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
aggregate_ops.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
gemm.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
meta_ops.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
ops.h - write 2 versions of new lstmLayer: one is based on own code, second uses mkl dnn api 2019-10-17 20:44:52 +03:00
random_ops.h Gamma and Poisson distributions (#27) 2019-11-04 15:42:28 +02:00
special_accumulation_ops.h [WIP] bunch of improvements (#257) 2019-09-11 20:12:09 +03:00
special_ops.h [WIP] bunch of improvements (#257) 2019-09-11 20:12:09 +03:00
special_random_ops.h [WIP] bunch of improvements (#257) 2019-09-11 20:12:09 +03:00
specials.h [WIP] CUDA tweaks (#60) 2019-07-20 23:06:48 +10:00
specials_cuda.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
specials_sparse.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00