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 7a2ac800dd
Nullify (#304) 
* initial commit

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

* bunch of tweaks

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

* hamming distance nullification

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

* Add output array value assignment for testing/debugging

Signed-off-by: Alex Black <blacka101@gmail.com>

* don't assign empty arrays

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

* conv2d/conv3d/depthwise2d nullified

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

* conv2d/conv3d/depthwise2d nullified

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

* conv2d/conv3d/depthwise2d nullified

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

* few more fixes

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

* im2col

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

* pooling?

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

* more nullified

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

* ismax nullified

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

* rollback ismax nullification

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

* synchronized cublas handle use on per-device basis

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

* hiding method from jcpp

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

* get rid of test assigns in DeclarableOp

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

* get rid of assigns

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

* proper deviceId is back

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

* include fixed

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

Co-authored-by: Alex Black <blacka101@gmail.com>
 		2020-03-20 08:49:28 +03:00
..
legacy libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
BarnesHutTsne.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
README.md Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
activations.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
addBias.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
adjust_hue.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
adjust_saturation.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
axis.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batched_gemm.cu Nullify (#304) 2020-03-20 08:49:28 +03:00
batchnorm.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
betaInc.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
col2im.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
compare_elem.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
concat.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
confusion.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
convolutions.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
cross.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
d_t_s.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
diGamma.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
diag.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
dilation2d.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
dropout.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
dynamic.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
extract_patches.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
fake_quantization.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
flatten.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
gather.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gather_nd.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gradient.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gru.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
hamming.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
hashcode.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
histogram.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
histogramFixedWidth.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
im2col.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
image_draw_bounding_boxes.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_resize.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_suppression.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
imagesHelpers.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
ismax.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
legacy_helper.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lgamma.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lrn.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstm.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstsq.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lup.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrixSetDiag.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
matrix_band.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrix_diag_part.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
max_pooling.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
maximum.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
merge.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
meshgrid.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
minimum.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
nth_element.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
one_hot.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
pad.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
percentile.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
polyGamma.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
prefix.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
print_variable.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
qr.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random_crop.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
range.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
reverse.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
roll.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
s_t_b.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
s_t_d.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
scatter.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
scatter_simple.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
scatter_update.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_max.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_mean.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_min.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_prod.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_sqrtn.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_sum.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sequence_mask.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sg_cb.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
shift.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
solve.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
split.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
sru.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
stack.cu profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
svd.cu profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
toggle_bits.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
top_k.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
transforms.cu [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
triangular_solve.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
weights.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
zeta.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00

README.md

This folder contains CUDA-specific implementations for operations.