Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
cavis/libnd4j/include/ops/declarable/helpers
History
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
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
BarnesHutTsne.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
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
matrixSetDiag.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_crop.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
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_common.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
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