cavis/libnd4j/include/ops/declarable/helpers/cuda
raver119 f990b2486d
simplified addBias2D for CUDA (#285)
Signed-off-by: raver119 <raver119@gmail.com>
2020-03-04 09:50:55 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
addBias.cu simplified addBias2D for CUDA (#285) 2020-03-04 09:50:55 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batchnorm.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +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 profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
confusion.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
convolutions.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
cross.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
s_t_d.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
scatter.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +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 profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
sru.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +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.