cavis/libnd4j/include/ops/declarable/helpers
raver119 efbfafe3f7
[WIP] gatherND fix (#176)
* one test for gather_nd

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

* get rid of old concat tests

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

* one printf

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

* one more legacy test removed

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

* gatherNd launch params fix

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

* gatherNd launch params fix

Signed-off-by: raver119 <raver119@gmail.com>
2019-08-27 12:35:14 +03:00
..
cpu [WIP] ops fixes (#168) 2019-08-26 19:37:05 +03:00
cuda [WIP] gatherND fix (#176) 2019-08-27 12:35:14 +03:00
impl [WIP] More tweaks (#173) 2019-08-27 10:37:10 +03:00
BarnesHutTsne.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
activations.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
addBias.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +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
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
bds.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 Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
cross.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +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 Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
flatten.h [WIP] More of CUDA (#63) 2019-07-20 23:15:14 +10: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
hashcode.h [WIP] More of CUDA operations (#69) 2019-07-20 23:20:41 +10:00
helpers.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +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_resize.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
image_suppression.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
ismax.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
legacy_helpers.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +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
lup.h [WIP] more CUDA stuff (#57) 2019-07-20 23:05:21 +10:00
matmul.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
matrixSetDiag.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
matrix_band.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
matrix_diag.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
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
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.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
segment_common.h [WIP] More of CUDA operations (#69) 2019-07-20 23:20:41 +10: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