cavis/libnd4j/include/ops/declarable/helpers/cuda
agibsonccc 968eaad2dd Update allowed data types in pooling 2021-02-07 19:53:55 +09:00
..
compression Update c++ copyrights 2021-02-01 21:31:45 +09:00
legacy Update c++ copyrights 2021-02-01 21:31:45 +09:00
BarnesHutTsne.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
README.md Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
activations.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
addBias.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
adjust_hue.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
adjust_saturation.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
axis.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
batched_gemm.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
batchnorm.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
betaInc.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
clip.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
col2im.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
compare_elem.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
concat.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
confusion.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
convolutions_col2vol.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
convolutions_conv2d.cu Split convolutions implementations for compilation speed up (#339) 2020-03-23 07:30:26 +03:00
convolutions_conv2dBP.cu Split convolutions implementations for compilation speed up (#339) 2020-03-23 07:30:26 +03:00
convolutions_depthwiseConv2d.cu Split convolutions implementations for compilation speed up (#339) 2020-03-23 07:30:26 +03:00
convolutions_depthwiseConv2dBP.cu Split convolutions implementations for compilation speed up (#339) 2020-03-23 07:30:26 +03:00
convolutions_pooling2d.cu Update allowed data types in pooling 2021-02-07 19:53:55 +09:00
convolutions_pooling2dBP.cu Update allowed data types in pooling 2021-02-07 19:53:55 +09:00
convolutions_pooling3d.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
convolutions_pooling3dBP.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
convolutions_sconv2d.cu Split convolutions implementations for compilation speed up (#339) 2020-03-23 07:30:26 +03:00
convolutions_upsampling2d.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
convolutions_upsampling2dBP.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
convolutions_upsampling3d.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
convolutions_upsampling3dBP.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
convolutions_vol2col.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
cross.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
d_t_s.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
diGamma.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
diag.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
dilation2d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
dropout.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
dynamic.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
extract_patches.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
fake_quantization.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
flatten.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
gather.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
gather_nd.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
gradient.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
hamming.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
hashcode.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
histogram.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
histogramFixedWidth.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
im2col.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
image_draw_bounding_boxes.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
image_resize.cu Eliminated error with resize implementation. (#418) 2020-05-27 21:15:03 +03:00
image_resize_v2.cu Eliminated error with resize implementation. (#418) 2020-05-27 21:15:03 +03:00
image_suppression.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
imagesHelpers.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
indexReductions.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
ismax.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
legacy_helper.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
lgamma.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lrn.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
lstm.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
lstsq.cu C++ rearrangements (#485) 2020-06-06 15:26:55 +03:00
lup.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
matrixSetDiag.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
matrix_band.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
matrix_diag_part.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
max_pooling.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
maximum.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
merge.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
meshgrid.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
minimum.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
nth_element.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
one_hot.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
pad.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
percentile.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
polyGamma.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
prefix.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
print_variable.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
qr.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
random.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
randomShuffle.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
random_crop.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
range.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
reverse.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
roll.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
s_t_b.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
s_t_d.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
scatter.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
scatter_simple.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
scatter_update.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
segment.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
segment_max.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
segment_mean.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
segment_min.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
segment_prod.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
segment_sqrtn.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
segment_sum.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
sequence_mask.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
sg_cb.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
shift.cu libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
solve.cu C++ rearrangements (#485) 2020-06-06 15:26:55 +03:00
split.cu Legacy API changes (#441) 2020-05-09 08:06:14 +03:00
sru.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
stack.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
svd.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
toggle_bits.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
top_k.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
transforms.cu Development updates (#9053) 2020-07-26 21:59:27 +09:00
triangular_solve.cu C++ rearrangements (#485) 2020-06-06 15:26:55 +03:00
updaterAdaDelta.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
updaterAdaGrad.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
updaterAdaMax.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
updaterAdam.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
updaterAmsGrad.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
updaterNadam.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
updaterNesterovs.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
updaterRmsProp.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
weights.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00
zeta.cu Update c++ copyrights 2021-02-01 21:31:45 +09:00

README.md

This folder contains CUDA-specific implementations for operations.