Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
cavis/libnd4j/include/ops/declarable/helpers
History
raver119 6de00bf75f
[WIP] Weekly update of repo (#8390) 
* [WIP] Fix compilation after nd4j changes (#37)

* Fix compilation.

* Some tests fixed

* Disable tests temporarily.

* Restored test

* Tests restored.

* Test restored.

* [WIP] perf tests (#40)

* special maxpool test

Signed-off-by: raver119 <raver119@gmail.com>

* special maxpool test

Signed-off-by: raver119 <raver119@gmail.com>

* Shyrma bnorm bp (#41)

Batchnorm backprop mkldnn

* 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>

* SameDiff op runtime benchmarking listener (#42)

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

* INLINE_LOOPS for windows

Signed-off-by: raver119 <raver119@gmail.com>

* [WIP] ThreadPool (#8)

This PR removes OpenMP use in 95% of cases
2019-11-13 17:15:18 +03:00
..
cpu
[WIP] Weekly update of repo (#8390)
2019-11-13 17:15:18 +03:00
cuda
[WIP] Weekly update of repo (#8390)
2019-11-13 17:15:18 +03:00
impl
[WIP] Weekly update of repo (#8390)
2019-11-13 17:15:18 +03:00
activations.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
addBias.h
[WIP] bunch of improvements (#257)
2019-09-11 20:12:09 +03:00
adjust_hue.h
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
adjust_saturation.h
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
axis.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
BarnesHutTsne.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
batched_gemm.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
batchnorm.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
betaInc.h
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
choose.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
col2im.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
compare_elem.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
confusion.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
convolutions.h
Platform helpers (#8216)
2019-09-11 21:50:28 +03:00
cross.h
[WIP] Weekly update of repo (#8390)
2019-11-13 17:15:18 +03:00
d_t_s.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
diag.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
dilation2d.h
[WIP] more CUDA stuff (#57)
2019-07-20 23:05:21 +10:00
dropout.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
dynamic.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
extract_patches.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
fake_quantization.h
Implementation of cuda kernel for fake_quant_with_min_max_vars_per_channels op.
2019-10-10 16:40:56 +03:00
flatten.h
[WIP] bunch of improvements (#257)
2019-09-11 20:12:09 +03:00
gather.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
gradient.h
[WIP] More of CUDA (#95)
2019-08-05 11:27:05 +10:00
gru.h
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
hamming.h
[WIP] bits_hamming_distance (#192)
2019-08-28 18:20:44 +03:00
hashcode.h
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
helpers.h
[WIP] Weekly update of repo (#8390)
2019-11-13 17:15:18 +03:00
histogram.h
[WIP] More CUDA fixes/updates (#62)
2019-07-20 23:07:42 +10:00
histogramFixedWidth.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
im2col.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
image_draw_bounding_boxes.h
Added helper declaration for draw_bounding_boxes op.
2019-10-04 21:16:34 +03:00
image_resize.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
image_suppression.h
Shugeo suppression overlaps (#9)
2019-10-30 13:43:45 +02:00
ismax.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
knn.h
Merge pull request #7 from KonduitAI/asto_nd4s_10172019
2019-10-23 12:11:25 +03:00
legacy_helpers.h
- add parameter alpha to elu and lrelu_bp (#213)
2019-08-31 20:57:39 +03:00
listdiff.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
lrn.h
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
lstm.h
[WIP] More of CUDA (#95)
2019-08-05 11:27:05 +10:00
lstmBlock.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
lstmLayer.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
lup.h
[WIP] more CUDA stuff (#57)
2019-07-20 23:05:21 +10:00
matmul.h
[WIP] Weekly update of repo (#8390)
2019-11-13 17:15:18 +03:00
matrix_band.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
matrix_diag_part.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
matrixSetDiag.h
Shyrma docs (#222)
2019-09-02 16:25:58 +03:00
max_pooling.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
meshgrid.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
minimax.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
multiUnique.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
nth_element.h
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
one_hot.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
percentile.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
polyGamma.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
prefix.h
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
random_crop.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
random.h
Shugeo random uniform int (#30)
2019-11-06 12:49:27 +02:00
range.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
reverse.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
rnn.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
roll.h
[WIP] Roll rewritten (#128)
2019-08-17 14:15:08 +03:00
s_t_b.h
[WIP] stb/bts nd (#144)
2019-08-21 21:11:46 +03:00
s_t_d.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
scatter.h
[WIP] more CUDA stuff (#57)
2019-07-20 23:05:21 +10:00
segment_common.h
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
segment.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
sequence_mask.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
sg_cb.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
shift.h
[WIP] right shift ops (#118)
2019-08-15 20:35:15 +03:00
sru.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
stack.h
[WIP] More of CUDA (#95)
2019-08-05 11:27:05 +10:00
svd.h
[WIP] more CUDA stuff (#57)
2019-07-20 23:05:21 +10:00
toggle_bits.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
top_k.h
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
transforms.h
[WIP] repeat op (#143)
2019-08-21 21:10:29 +03:00
unique.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
weights.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
where.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
zeta.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00