cavis/libnd4j/include/ops/declarable/helpers
Oleh b686368b82
Refactoring split operation (#266)
* libnd4j moved split operation implementation to helpers before special case adding

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j minor fixes for general split operation move, merge master

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libndj4 split cpu implementation

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* - provide cuda helper for split op

Signed-off-by: Yurii <iuriish@yahoo.com>

* - minor correction

Signed-off-by: Yurii <iuriish@yahoo.com>

* - minor correction 2

Signed-off-by: Yurii <iuriish@yahoo.com>

* libnd4j moved split implementation from specials to split.cpp

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j update loopkind selections for 3D, 4D and 5D cases

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j removed unnecessary BUILD_SINGLE_TEMPLATE

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

Co-authored-by: Yurii Shyrma <iuriish@yahoo.com>
2020-02-26 10:20:39 +03:00
..
cpu Refactoring split operation (#266) 2020-02-26 10:20:39 +03:00
cuda Split operation improvement (#262) 2020-02-24 08:22:41 +03:00
impl Shugeo unordered map (#256) 2020-02-24 07:51:01 +03:00
BarnesHutTsne.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
activations.h [WIP] CUDA tests (#95) 2019-12-02 21:37:21 +03:00
addBias.h [WIP] bunch of improvements (#257) 2019-09-11 20:12:09 +03:00
adjust_hue.h [WIP] Oleh rgb yuv (#147) 2019-12-24 18:30:54 +03: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
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 [WIP] CUDA tests (#95) 2019-12-02 21:37:21 +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 cuDNN integration (#150) 2020-01-20 21:32:46 +03:00
crop_and_resize.h More compilation units (#246) 2020-02-17 10:23:05 +03:00
cross.h Minor improvements (#255) 2020-02-20 11:43:26 +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
gammaMathFunc.h Shyrma adjust (#98) 2019-12-03 09:40:45 +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 String changes (#3) 2020-01-04 13:27:50 +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 [WIP] CUDA tests (#95) 2019-12-02 21:37:21 +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 More compilation units (#246) 2020-02-17 10:23:05 +03:00
image_suppression.h Shugeo_release_fixes3 (#81) 2019-11-28 21:08:51 +03:00
imagesHelpers.h [WIP] Oleh rgb yuv (#147) 2019-12-24 18:30:54 +03: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
lgamma.h Shugeo lgamma (#170) 2020-01-20 12:29:36 +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 Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
lstmBlock.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
lstmLayer.h Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
lup.h [WIP] Shugeo lup (#126) 2019-12-20 17:56:28 +03:00
matmul.h [WIP] ThreadPool (#8) 2019-11-13 17:04:59 +03:00
matrixSetDiag.h Shyrma docs (#222) 2019-09-02 16:25:58 +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
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 [WIP] CUDA tests (#95) 2019-12-02 21:37:21 +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
prefix.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
print_variable.h String changes (#3) 2020-01-04 13:27:50 +03:00
qr.h Shugeo qr (#153) 2020-01-22 13:59:36 +03:00
random.h Oleh multinomial (#163) 2020-01-06 22:35:05 +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 Shugeo image resize bicubic (#56) 2019-11-20 21:11:04 +02:00
s_t_d.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
scatter.h Shyrma scatter (#84) 2019-11-26 20:29:09 +03: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
solve.h Shugeo solve linear (#191) 2020-02-04 08:59:11 +03:00
sparse_to_dense.h String changes (#3) 2020-01-04 13:27:50 +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 Split operation improvement (#262) 2020-02-24 08:22:41 +03:00
triangular_solve.h Shugeo solve triangular (#173) 2020-01-22 10:48:03 +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