cavis/libnd4j/include/ops/declarable/generic
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
..
activations - add parameter alpha to elu and lrelu_bp (#213) 2019-08-31 20:57:39 +03:00
bitwise - bits_hamming_distance dtype fix (#8208) 2019-09-06 08:59:05 +03:00
blas [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
boolean Small fixes (#43) 2019-07-20 22:19:20 +10:00
broadcastable Added igamma and igammac broadcastable ops implementations and tests. 2019-10-16 14:02:53 +03:00
datatypes Refactored buffer copying to avoid wrong usage of buffers. 2019-10-02 16:51:09 +03:00
flow Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
grad Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
helpers Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
kernels Merge pull request #7 from KonduitAI/asto_nd4s_10172019 2019-10-23 12:11:25 +03:00
list Shugeo cuda tests (#116) 2019-08-15 13:54:47 +03:00
loss [WIP] more CUDA stuff (#57) 2019-07-20 23:05:21 +10:00
nlp [WIP] Error handling (#169) 2019-08-26 19:57:51 +03:00
nn Add SameDiff memory reuse memory manager (array cache) (#39) 2019-11-12 21:15:44 +11:00
parity_ops Shugeo strided slice bp fix2 (#33) 2019-11-07 13:44:02 +03:00
random - new NDArrayFactory scalar constructor 2019-11-08 08:49:41 +03:00
recurrent Shyrma concat (#24) 2019-11-03 12:37:19 +02:00
shape [WIP] size etc (#155) 2019-08-23 12:31:12 +03:00
tests Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
thrid_party [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
transforms Op Fixes (#28) 2019-11-05 00:05:04 +11:00
tsne [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
CustomOperations.cpp Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
README.md Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
parity_ops.cpp Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00

README.md

Generic ops

Put any cross platform operations here. Anything platform specific should be put in: in ops/helpers.