| .. |
|
argmax_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
argmax_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
assign_add_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
assign_add_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
batchnorm_fold2_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
batchnorm_fold2_impl.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
batchnorm_fold_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
batchnorm_fold_impl.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
concatv2_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
concatv2_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
correction_mul_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
correction_mul_impl.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
cross_entropy_impl.cu
|
fix bug in cross entropy error
|
2020-04-21 14:21:46 +08:00 |
|
cross_entropy_impl.cuh
|
fix bug in cross entropy error
|
2020-04-21 14:21:46 +08:00 |
|
dropout_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
dropout_impl.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
equalcount_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
equalcount_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
fake_quant_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
fake_quant_impl.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
fake_quant_per_channel_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
fake_quant_per_channel_impl.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
float_status_impl.cu
|
gpu add float_status kernel
|
2020-04-20 19:44:40 +08:00 |
|
float_status_impl.cuh
|
gpu add float_status kernel
|
2020-04-20 19:44:40 +08:00 |
|
gather.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
gather.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
momentum_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
momentum_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
one_hot_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
one_hot_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
pad_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
pad_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
select_impl.cu
|
gpu add kernel select
|
2020-04-14 20:03:33 +08:00 |
|
select_impl.cuh
|
gpu add kernel select
|
2020-04-14 20:03:33 +08:00 |
|
slice_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
slice_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
sparse_cross_entropy_cuda_impl.cu
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
sparse_cross_entropy_cuda_impl.cuh
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
transpose_impl.cu
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
transpose_impl.cuh
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
unary_op_impl.cu
|
gpu dynamic memory pool supports multi-allReduce
|
2020-04-22 15:43:45 +08:00 |
|
unary_op_impl.cuh
|
gpu dynamic memory pool supports multi-allReduce
|
2020-04-22 15:43:45 +08:00 |
2891f0d20d