Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
cavis/libnd4j/include/ops/declarable/helpers/cuda
History
raver119 aceb915557
[WIP] tests fixes (#130) 
* no openmp for ClipByGlobalNorm

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

* one more bfloat16 rng test

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

* assertion fix

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

* - legacy IsMax gone
- linear IsMax gets shapeInfo argument

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

* get rid of legacy IsMax tests

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

* IsMax is custom op now

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

* more blocks for ismax

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

* one more test

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

*  - sqrt test
 - some legacy code removed from CudaExecutioner
 - Transforms.asin tweaks

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

* - TransformFloat fix

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

* - ismax fix
- SpaceToBatchND/BatchToSpaceND wrappers
- couple of legacy tests removed

Signed-off-by: raver119 <raver119@gmail.com>
2019-08-19 11:33:15 +03:00
..
legacy
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
activations.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
adjust_hue.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
adjust_saturation.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
axis.cu
[WIP] Roll rewritten (#128)
2019-08-17 14:15:08 +03:00
BarnesHutTsne.cu
[WIP] More of CUDA (#95)
2019-08-05 11:27:05 +10:00
batched_gemm.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
batchnorm.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
bds.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +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] multi-device support (#80)
2019-08-14 16:52:34 +03:00
compare_elem.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
concat.cu
Shyrma concat2 (#119)
2019-08-16 06:57:20 +03:00
confusion.cu
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
convolutions.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
cross.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
d_t_s.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
diag.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
dilation2d.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
dropout.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
dynamic.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +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] More of CUDA (#63)
2019-07-20 23:15:14 +10:00
gather_nd.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
gather.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
gradient.cu
build fix
2019-08-05 11:27:05 +10:00
gru.cu
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
hashcode.cu
[WIP] More of CUDA (#95)
2019-08-05 11:27:05 +10:00
histogram.cu
Shugeo cuda tests (#116)
2019-08-15 13:54:47 +03:00
histogramFixedWidth.cu
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
im2col.cppc
Fix OpenMP by replacing pragmas with macros (#8026)
2019-07-18 20:13:56 +09:00
im2col.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
image_resize.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
image_suppression.cu
Shugeo cuda tests (#116)
2019-08-15 13:54:47 +03:00
ismax.cu
[WIP] tests fixes (#130)
2019-08-19 11:33:15 +03:00
legacy_helper.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
lrn.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
lstm.cu
[WIP] More of CUDA (#95)
2019-08-05 11:27:05 +10:00
lup.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +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
matrix_diag.cu
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
matrixSetDiag.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
Shugeo cuda tests (#116)
2019-08-15 13:54:47 +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] multi-device support (#80)
2019-08-14 16:52:34 +03:00
one_hot.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
pad.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
percentile.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
polyGamma.cu
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
prefix.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
random_crop.cu
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
range.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
README.md
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
reverse.cu
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
roll.cu
[WIP] Roll rewritten (#128)
2019-08-17 14:15:08 +03:00
s_t_b.cu
[WIP] More of CUDA (#95)
2019-08-05 11:27:05 +10:00
s_t_d.cu
Merge master to upstream (#7945)
2019-06-27 18:37:04 +03:00
scatter_simple.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
scatter_update.cu
[WIP] build time improvements (#106)
2019-08-07 17:49:13 +03:00
scatter.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
segment_max.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
segment_mean.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
segment_min.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
segment_prod.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
segment_sqrtn.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
segment_sum.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
segment.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
sequence_mask.cu
[WIP] More of CUDA operations (#69)
2019-07-20 23:20:41 +10:00
sg_cb.cu
Shugeo cuda tests (#116)
2019-08-15 13:54:47 +03:00
shift.cu
[WIP] right shift ops (#118)
2019-08-15 20:35:15 +03:00
sru.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
stack.cu
temporary stack fix
2019-08-05 11:28:07 +10:00
svd.cu
[WIP] more CUDA stuff (#57)
2019-07-20 23:05:21 +10:00
toggle_bits.cu
Shugeo cuda tests (#116)
2019-08-15 13:54:47 +03:00
top_k.cu
[WIP] multi-device support (#80)
2019-08-14 16:52:34 +03:00
transforms.cu
no openmp for ClipByGlobalNorm
2019-08-15 18:24:29 +03:00
weights.cu
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00
zeta.cu
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +03:00

README.md

This folder contains CUDA-specific implementations for operations.