* linear equations systems solve op. Initial commit. Signed-off-by: shugeo <sgazeos@gmail.com> * Fixed compiling issues. Signed-off-by: shugeo <sgazeos@gmail.com> * Linear equations systems solve. The next stage commit. Signed-off-by: shugeo <sgazeos@gmail.com> * Added test for linear equations systems solve operation. Signed-off-by: shugeo <sgazeos@gmail.com> * Added additional test and fixed lower matrix retrievance. * Implementation for solve of the systems of linear equations." Signed-off-by: shugeo <sgazeos@gmail.com> * Refactored permutation generation. Signed-off-by: shugeo <sgazeos@gmail.com> * Added restore for permutations batched with cuda helper for solve op. Signed-off-by: shugeo <sgazeos@gmail.com> * Finished cuda implementation for solve op helpers. Signed-off-by: shugeo <sgazeos@gmail.com> * Refactored cpu helpers for solve op. Signed-off-by: shugeo <sgazeos@gmail.com> * Fix gtest output on Windows * Fixed issue with permutation matrix for cuda implementation. Signed-off-by: shugeo <sgazeos@gmail.com> * Fixed issue with permutation matrix for cpu implementation. Signed-off-by: shugeo <sgazeos@gmail.com> * Eliminated waste comments. Signed-off-by: shugeo <sgazeos@gmail.com> * LinearSolve added * Mapping added * Javadoc added * Refactored implementation of triangular_solve helpers and tests for solve matrix equations generally. Signed-off-by: shugeo <sgazeos@gmail.com> * Added a test for solve op. Signed-off-by: shugeo <sgazeos@gmail.com> * Solve test added * Fix for TF import Co-authored-by: Serhii Shepel <9946053+sshepel@users.noreply.github.com> Co-authored-by: raver119 <raver119@gmail.com> Co-authored-by: Alexander Stoyakin <alexander.stoyakin@gmail.com> |
||
---|---|---|
.. | ||
legacy | ||
BarnesHutTsne.cu | ||
README.md | ||
activations.cu | ||
addBias.cu | ||
adjust_hue.cu | ||
adjust_saturation.cu | ||
axis.cu | ||
batched_gemm.cu | ||
batchnorm.cu | ||
betaInc.cu | ||
col2im.cu | ||
compare_elem.cu | ||
concat.cu | ||
confusion.cu | ||
convolutions.cu | ||
cross.cu | ||
d_t_s.cu | ||
diGamma.cu | ||
diag.cu | ||
dilation2d.cu | ||
dropout.cu | ||
dynamic.cu | ||
extract_patches.cu | ||
fake_quantization.cu | ||
flatten.cu | ||
gather.cu | ||
gather_nd.cu | ||
gradient.cu | ||
gru.cu | ||
hamming.cu | ||
hashcode.cu | ||
histogram.cu | ||
histogramFixedWidth.cu | ||
im2col.cu | ||
image_draw_bounding_boxes.cu | ||
image_resize.cu | ||
image_suppression.cu | ||
imagesHelpers.cu | ||
ismax.cu | ||
legacy_helper.cu | ||
lgamma.cu | ||
lrn.cu | ||
lstm.cu | ||
lup.cu | ||
matrixSetDiag.cu | ||
matrix_band.cu | ||
matrix_diag_part.cu | ||
max_pooling.cu | ||
maximum.cu | ||
merge.cu | ||
meshgrid.cu | ||
minimum.cu | ||
nth_element.cu | ||
one_hot.cu | ||
pad.cu | ||
percentile.cu | ||
polyGamma.cu | ||
prefix.cu | ||
print_variable.cu | ||
qr.cu | ||
random.cu | ||
random_crop.cu | ||
range.cu | ||
reverse.cu | ||
roll.cu | ||
s_t_b.cu | ||
s_t_d.cu | ||
scatter.cu | ||
scatter_simple.cu | ||
scatter_update.cu | ||
segment.cu | ||
segment_max.cu | ||
segment_mean.cu | ||
segment_min.cu | ||
segment_prod.cu | ||
segment_sqrtn.cu | ||
segment_sum.cu | ||
sequence_mask.cu | ||
sg_cb.cu | ||
shift.cu | ||
solve.cu | ||
sru.cu | ||
stack.cu | ||
svd.cu | ||
toggle_bits.cu | ||
top_k.cu | ||
transforms.cu | ||
triangular_solve.cu | ||
weights.cu | ||
zeta.cu |
README.md
This folder contains CUDA-specific implementations for operations.