cavis/libnd4j/include/ops/declarable/helpers
Abdelrauf 69d91e272a
- new implementations for Index Reductions (#421)
* - new implementations for Index Reductions
- small fix in the legacy reduction
- disabled index reduction bench tests inside Playground

Signed-off-by: Abdelrauf <rauf@konduit.ai>

* Allow LIBND4J_TYPES

Signed-off-by: Abdelrauf <rauf@konduit.ai>

* index reduction stuff split into bunch of units

* meh

* IMax switched to new impl

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

* minor fix + test

* minor fix

* index range fix

Signed-off-by: Abdelrauf <rauf@konduit.ai>

* noop on empty outputs

* minor fix

* minor fix

Signed-off-by: Abdelrauf <rauf@konduit.ai>

* ArgMax replaces IMax

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

* argmax/argmin/argamax/argamin shape functions updated

* ArgAmax/ArgAmin/ArgMin replaces IAMax/IAMin/IMin

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

* argmax/argmin/argamax/argamin CUDA

* IMax replaced in dl4j

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

* Codegen output

* imports fixed

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

* fix compilation issue

Signed-off-by: Abdelrauf <rauf@konduit.ai>

* Auto-generate compilation units

Signed-off-by: Abdelrauf <rauf@konduit.ai>

* Should fix NDArray refactored function calls in indexReductions.cu

Signed-off-by: Abdelrauf <rauf@konduit.ai>

Co-authored-by: raver119@gmail.com <raver119@gmail.com>
Co-authored-by: Alexander Stoyakin <alexander.stoyakin@gmail.com>
2020-05-14 13:41:55 +03:00
..
cpu - new implementations for Index Reductions (#421) 2020-05-14 13:41:55 +03:00
cuda - new implementations for Index Reductions (#421) 2020-05-14 13:41:55 +03:00
impl Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
BarnesHutTsne.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
activations.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
addBias.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
adjust_hue.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
adjust_saturation.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
axis.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batched_gemm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batchnorm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
betaInc.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
choose.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
col2im.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
compare_elem.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
compression.h Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
confusion.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
convolutions.h Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
crop_and_resize.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
cross.h Revert "OpenMP Threads execution (#297)" (#299) 2020-03-09 08:22:49 +03:00
d_t_s.h Fix for certain non-ews cases (#402) 2020-04-21 12:41:30 +03:00
diag.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
dilation2d.h Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
dropout.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
dynamic.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
extract_patches.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
fake_quantization.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
flatten.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gammaMathFunc.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gather.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gradient.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gru.h Shyrma gru bp (#377) 2020-04-16 08:09:04 +03:00
hamming.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
hashcode.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
helpers.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
histogram.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
histogramFixedWidth.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
im2col.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_draw_bounding_boxes.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_resize.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_suppression.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
imagesHelpers.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
ismax.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
knn.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
legacy_helpers.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lgamma.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
listdiff.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lrn.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmBlock.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmLayer.h Shyrma gru bp (#377) 2020-04-16 08:09:04 +03:00
lstsq.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lup.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matmul.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrixSetDiag.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrix_band.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrix_diag_part.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
max_pooling.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
meshgrid.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
minimax.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
multiUnique.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
nth_element.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
one_hot.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
percentile.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
prefix.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
print_variable.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
qr.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random_crop.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
range.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
reductions.h - new implementations for Index Reductions (#421) 2020-05-14 13:41:55 +03:00
reverse.h Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
rnn.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
roll.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
s_t_b.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
s_t_d.h Fix for certain non-ews cases (#402) 2020-04-21 12:41:30 +03:00
scatter.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_common.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sequence_mask.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sg_cb.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
shift.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
solve.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sparse_to_dense.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sru.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
stack.h profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
svd.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
threshold.h compression ops (#436) 2020-05-08 20:59:39 +03:00
toggle_bits.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
top_k.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
transforms.h Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
triangular_solve.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
unique.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
updatersHelpers.h Learning updaters for gradient (#335) 2020-03-23 07:28:31 +03:00
weights.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
where.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
zeta.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00