cavis/libnd4j/include/ops/declarable/helpers
shugeo 0eca33ad94
Shugeo cuda solver fix (#383)
* Refactored cuSolver handle usage to handle LaunchContext instance properly.

Signed-off-by: shugeo <sgazeos@gmail.com>

* Refactored svd solver usage with LaunchContext instance singleton.

Signed-off-by: shugeo <sgazeos@gmail.com>

* add device locks for cuSolver uses

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

Co-authored-by: raver119 <raver119@gmail.com>
2020-04-17 16:52:08 +03:00
..
cpu Shyrma gru bp (#377) 2020-04-16 08:09:04 +03:00
cuda Shugeo cuda solver fix (#383) 2020-04-17 16:52:08 +03:00
impl Shyrma gru bp (#377) 2020-04-16 08:09:04 +03:00
BarnesHutTsne.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
activations.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
addBias.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
adjust_hue.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
adjust_saturation.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
axis.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batched_gemm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batchnorm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
betaInc.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
choose.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
col2im.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
compare_elem.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
confusion.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
convolutions.h Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
crop_and_resize.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
cross.h Revert "OpenMP Threads execution (#297)" (#299) 2020-03-09 08:22:49 +03:00
d_t_s.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
diag.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
dilation2d.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
dropout.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
dynamic.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
extract_patches.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
fake_quantization.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
flatten.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gammaMathFunc.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gather.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gradient.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gru.h Shyrma gru bp (#377) 2020-04-16 08:09:04 +03:00
hamming.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
hashcode.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
helpers.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
histogram.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
histogramFixedWidth.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
im2col.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_draw_bounding_boxes.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_resize.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
image_suppression.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
imagesHelpers.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
ismax.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
knn.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
legacy_helpers.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lgamma.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
listdiff.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lrn.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmBlock.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmLayer.h Shyrma gru bp (#377) 2020-04-16 08:09:04 +03:00
lstsq.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lup.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matmul.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrixSetDiag.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrix_band.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matrix_diag_part.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
max_pooling.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
meshgrid.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
minimax.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
multiUnique.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
nth_element.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
one_hot.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
percentile.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
prefix.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
print_variable.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
qr.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random_crop.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
range.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
reverse.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
rnn.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
roll.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
s_t_b.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
s_t_d.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
scatter.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
segment_common.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sequence_mask.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sg_cb.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
shift.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
solve.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sparse_to_dense.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
sru.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
stack.h profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
svd.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
toggle_bits.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
top_k.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
transforms.h Backpropagation implementation of mergemax, mergeadd and mergeavg ops (#343) 2020-03-25 08:40:30 +03:00
triangular_solve.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
unique.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
updatersHelpers.h Learning updaters for gradient (#335) 2020-03-23 07:28:31 +03:00
weights.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
where.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
zeta.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00