cavis/libnd4j/include/ops/declarable/helpers/cuda
Yurii Shyrma cb4c9377b1 Shyrma docs (#222)
* - documenting and profiling matrix_set_diag cuda kernel

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

* - correct formula of pnorm pooling in cuda 2d/3d kernels
- remove helper matrix_diag which duplicates work of helper matrix_set_diag

Signed-off-by: Yurii <yurii@skymind.io>
2019-09-02 16:25:58 +03:00
..
legacy - add parameter alpha to elu and lrelu_bp (#213) 2019-08-31 20:57:39 +03:00
BarnesHutTsne.cu [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
README.md Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
activations.cu Shyrma softmax (#209) 2019-08-30 20:31:05 +03:00
adjust_hue.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
adjust_saturation.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
axis.cu [WIP] Roll rewritten (#128) 2019-08-17 14:15:08 +03:00
batched_gemm.cu Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
batchnorm.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
betaInc.cu Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
col2im.cppc Fix OpenMP by replacing pragmas with macros (#8026) 2019-07-18 20:13:56 +09:00
col2im.cu [WIP] stb/bts nd (#144) 2019-08-21 21:11:46 +03:00
compare_elem.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
concat.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
confusion.cu [WIP] confusion (#180) 2019-08-27 14:30:37 +03:00
convolutions.cu Shyrma docs (#222) 2019-09-02 16:25:58 +03:00
cross.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
d_t_s.cu Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
diag.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
dilation2d.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
dropout.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
dynamic.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
extract_patches.cu Shugeo cuda cuda (#105) 2019-08-07 15:29:17 +03:00
fake_quantization.cu Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
flatten.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
gather.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
gather_nd.cu [WIP] gatherND fix (#176) 2019-08-27 12:35:14 +03:00
gradient.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
gru.cu [WIP] More of CUDA operations (#69) 2019-07-20 23:20:41 +10:00
hamming.cu [WIP] bits_hamming_distance (#192) 2019-08-28 18:20:44 +03:00
hashcode.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
histogram.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
histogramFixedWidth.cu [WIP] More fixes (#178) 2019-08-27 13:21:01 +03:00
im2col.cppc Fix OpenMP by replacing pragmas with macros (#8026) 2019-07-18 20:13:56 +09:00
im2col.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
image_resize.cu Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
image_suppression.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
ismax.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
legacy_helper.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
lrn.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
lstm.cu [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
lup.cu lup context fix (#164) 2019-08-24 16:57:48 +03:00
matrixSetDiag.cu Shyrma docs (#222) 2019-09-02 16:25:58 +03:00
matrix_band.cu Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
matrix_diag_part.cu Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
max_pooling.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
maximum.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
merge.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
meshgrid.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
minimum.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
nth_element.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
one_hot.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
pad.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
percentile.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
polyGamma.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
prefix.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
random_crop.cu Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
range.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
reverse.cu [WIP] reverse_sequence (#188) 2019-08-28 11:14:22 +03:00
roll.cu [WIP] Roll rewritten (#128) 2019-08-17 14:15:08 +03:00
s_t_b.cu [WIP] stb/bts nd (#144) 2019-08-21 21:11:46 +03:00
s_t_d.cu Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
scatter.cu [WIP] ops fixes (#168) 2019-08-26 19:37:05 +03:00
scatter_simple.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
scatter_update.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
segment.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
segment_max.cu [WIP] More fixes (#190) 2019-08-28 15:38:57 +03:00
segment_mean.cu [WIP] few more fixes (#182) 2019-08-27 21:00:38 +03:00
segment_min.cu [WIP] More fixes (#190) 2019-08-28 15:38:57 +03:00
segment_prod.cu [WIP] More fixes (#190) 2019-08-28 15:38:57 +03:00
segment_sqrtn.cu Shugeo segment fix2 (#185) 2019-08-27 18:25:39 +03:00
segment_sum.cu [WIP] More fixes (#190) 2019-08-28 15:38:57 +03:00
sequence_mask.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
sg_cb.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
shift.cu [WIP] right shift ops (#118) 2019-08-15 20:35:15 +03:00
sru.cu syncthreads (#136) 2019-08-20 18:28:43 +03:00
stack.cu temporary stack fix 2019-08-05 11:28:07 +10:00
svd.cu [WIP] More fixes (#190) 2019-08-28 15:38:57 +03:00
toggle_bits.cu [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
top_k.cu [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
transforms.cu one more tiny cuda fux 2019-09-02 11:49:13 +03:00
weights.cu Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
zeta.cu [WIP] more fixes (#159) 2019-08-23 19:20:50 +03:00

README.md

This folder contains CUDA-specific implementations for operations.