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