cavis/libnd4j/include/ops/declarable/helpers
raver119 c969b724bb [WIP] more CUDA stuff (#57)
* initial commit

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

* Added gradcheck test for dynamic_partition_bp op.

* - implementation of dilation op (cpu and cuda)

Signed-off-by: Yurii <yurii@skymind.io>

* Fixed broadcast_dynamic_shape 1D case and tests.

* Fixed usage of default integer arguments.

* Fixed dynamic_partition_bp op and tests.

* Eliminated test with grad check for dynamic_partition_bp op.

* start working on cuda svd - porting available corresponding api from cuSOLVER library

Signed-off-by: Yurii <yurii@skymind.io>

* provide prelu_bp

Signed-off-by: Yurii <yurii@skymind.io>

* - provide gruCell_bp (old version ??)

Signed-off-by: Yurii <yurii@skymind.io>

* - polishing cumsum_bp and cumprod_bp tests

Signed-off-by: Yurii <yurii@skymind.io>

* provide sparseSoftmaxCrossEntropyWithLogits and sparseSoftmaxCrossEntropyWithLogits_grad

Signed-off-by: Yurii <yurii@skymind.io>

* Fixed atomicMul with float input/output

* implementation of cuda kernel for triu_bp operation

Signed-off-by: Yurii <yurii@skymind.io>

* Refactored lup helper to add parrallel computing.

* cusolver libraries

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

* uncomment cuSolver APIs in svd.cu

Signed-off-by: Yurii <yurii@skymind.io>

* cusolver var

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

* - further work on cuSolver svd

Signed-off-by: Yurii <yurii@skymind.io>

* Implement usage of cuda solver to LUP decomposition.

* - correct naames in lup functions

Signed-off-by: Yurii <yurii@skymind.io>

* correct svdQR cuda

Signed-off-by: Yurii <yurii@skymind.io>

* - provide transpositions of input matrices in case of c order in svdCudaQR

Signed-off-by: Yurii <yurii@skymind.io>

* Fixed implementation issues with LUP usign cuda solver.

* Implementation of matrix_determinant helper with cuda kernels. Working revision.

* Implemented log_matrix_determinant helper with cuda kernels.

* - implementation of batched cuda svd

Signed-off-by: Yurii <yurii@skymind.io>

* Refactored cholesky helper and implementation of cuda solver cholesky batch.

* - implementation of cuda kernel for tile bp

Signed-off-by: Yurii <yurii@skymind.io>

* Implementation of cholesky and logdet with cuda kernels.

* - implementation of cuda kernel for sru_bidirectional

Signed-off-by: Yurii <yurii@skymind.io>

* Fixed cholesky helper.

* Cholesky op helper implementation. Working double-based cublas implementation.

* bad import excluded

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

* Finished with cuda implementation of cholesky helper and tests.

* - implementation of cuda kernel for sru_bidirectional_backprop operation

Signed-off-by: Yurii <yurii@skymind.io>

* Implementation of matrix_inverse op helper with cuda kernels. The first revision.

* - start working on gruCell_bp

Signed-off-by: Yurii <yurii@skymind.io>

* Implementation of matrix_inverse helper.

* - further work on new gruCell_bp

Signed-off-by: Yurii <yurii@skymind.io>

* cuBLAS related fixes

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

* calculateOutputShapes() now passes device buffers as well

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

* special concat/average/accumulate init host pointers now

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

* few more tweaks

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

* additional CudaDataBufferFactory signatures certain for data types

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

* cuSolver host buffer

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

* buffer to buffer memcpy host ptr allocation

Signed-off-by: raver119 <raver119@gmail.com>
2019-07-20 23:05:21 +10:00
..
cpu [WIP] more CUDA stuff (#57) 2019-07-20 23:05:21 +10:00
cuda [WIP] more CUDA stuff (#57) 2019-07-20 23:05:21 +10:00
impl Add libnd4j benchmarks (#3) 2019-07-20 22:54:44 +10: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 Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
adjust_saturation.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03: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 Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
gather.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
gru.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
hashcode.h [WIP] INDArray hashCode() impl (#50) 2019-07-20 22:22:11 +10:00
helpers.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03: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 Merge master to upstream (#7945) 2019-06-27 18:37:04 +03: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 Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
s_t_b.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +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
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
sru.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
stack.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03: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 Merge master to upstream (#7945) 2019-06-27 18:37:04 +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