* Implementation for non_max_suppression_v3 was added. Initial version * Added check for overcome threshold. * Added definition for V3 method. * java remapping for NonMaxSuppressionV3 Signed-off-by: raver119 <raver119@gmail.com> * Fixed proporly processing of an empty output and test. * Refactored op to less threshold data to float. * Implemented cuda-based helper for non_max_suppression_v3 op. * Fixed fake_quant_with_min_max_vars op. * Fixed tests with float numbers. * - assert now stops execution - sortByKey/sortByValue now have input validation Signed-off-by: raver119 <raver119@gmail.com> * missing var Signed-off-by: raver119 <raver119@gmail.com> * Fixed proper processing for zero max_size inputs. * Refactored kernel callers. * Fixed return statement for logdet op helper. * Refactored unsorted segment SqrtN op. * get back 8 tail bytes on CUDA Signed-off-by: raver119 <raver119@gmail.com> * Refactored segment prod ops and helpers for cuda and tests. * Additional test. * CudaWorkspace tests updated for 8 tail bytes Signed-off-by: raver119 <raver119@gmail.com> * special atomic test Signed-off-by: raver119 <raver119@gmail.com> * atomicMul/atomicDiv fix for 16bit values Signed-off-by: raver119 <raver119@gmail.com> * Eliminated waste prints. |
||
---|---|---|
.. | ||
BarnesHutTsne.cpp | ||
README.md | ||
activations.cpp | ||
addBias.cpp | ||
adjust_hue.cpp | ||
adjust_saturation.cpp | ||
axis.cpp | ||
batched_gemm.cpp | ||
batchnorm.cpp | ||
betaInc.cpp | ||
col2im.cpp | ||
compare_elem.cpp | ||
confusion.cpp | ||
convolutions.cpp | ||
cross.cpp | ||
d_t_s.cpp | ||
diag.cpp | ||
dilation2d.cpp | ||
dropout.cpp | ||
dynamic.cpp | ||
extract_patches.cpp | ||
fake_quantization.cpp | ||
flatten.cpp | ||
gather.cpp | ||
gradient.cpp | ||
gru.cpp | ||
hamming.cpp | ||
hashcode.cpp | ||
histogram.cpp | ||
histogramFixedWidth.cpp | ||
im2col.cpp | ||
image_draw_bounding_boxes.cpp | ||
image_resize.cpp | ||
image_suppression.cpp | ||
ismax.cpp | ||
legacy_helper.cpp | ||
lrn.cpp | ||
lstm.cpp | ||
lup.cpp | ||
matrixSetDiag.cpp | ||
matrix_band.cpp | ||
matrix_diag_part.cpp | ||
max_pooling.cpp | ||
meshgrid.cpp | ||
minimax.cpp | ||
nth_element.cpp | ||
one_hot.cpp | ||
percentile.cpp | ||
polyGamma.cpp | ||
prefix.cpp | ||
random.cpp | ||
random_crop.cpp | ||
range.cpp | ||
reverse.cpp | ||
roll.cpp | ||
s_t_b.cpp | ||
s_t_d.cpp | ||
scatter.cpp | ||
segment.cpp | ||
sequence_mask.cpp | ||
sg_cb.cpp | ||
shift.cpp | ||
sru.cpp | ||
stack.cpp | ||
svd.cpp | ||
toggle_bits.cpp | ||
top_k.cpp | ||
transforms.cpp | ||
weights.cpp | ||
zeta.cpp |
README.md
This folder contains OpenMP implementations for operations helpers. Basically suited for homogenous x86-like platforms.