| .. |
|
bias_add_grad_gpu_kenel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
bias_add_grad_gpu_kenel.h
|
fix cudnn type error
|
2020-04-23 11:14:41 +08:00 |
|
conv2d_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
conv2d_gpu_kernel.h
|
!677 GPU fix codex for conv2d
|
2020-04-26 14:40:26 +08:00 |
|
conv2d_grad_filter_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
conv2d_grad_filter_gpu_kernel.h
|
!677 GPU fix codex for conv2d
|
2020-04-26 14:40:26 +08:00 |
|
conv2d_grad_input_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
conv2d_grad_input_gpu_kernel.h
|
!677 GPU fix codex for conv2d
|
2020-04-26 14:40:26 +08:00 |
|
dropout_gpu_kernel.cc
|
clear the warmming scan by package
|
2020-04-15 21:45:06 +08:00 |
|
dropout_gpu_kernel.h
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
dropout_grad_kernel.cc
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
dropout_grad_kernel.h
|
add quantizaiton gpu op
|
2020-04-13 16:51:41 +08:00 |
|
flatten_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
flatten_gpu_kernel.h
|
update some ops
|
2020-04-21 12:32:10 +08:00 |
|
flatten_grad_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
flatten_grad_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
fused_batch_norm_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
fused_batch_norm_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
fused_batchnorm_grad_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
fused_batchnorm_grad_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
lstm_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
lstm_gpu_kernel.h
|
remove make_unique.h
|
2020-04-07 14:00:09 -04:00 |
|
lstm_grad_data_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
lstm_grad_data_gpu_kernel.h
|
remove make_unique.h
|
2020-04-07 14:00:09 -04:00 |
|
lstm_grad_weight_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
lstm_grad_weight_gpu_kernel.h
|
remove make_unique.h
|
2020-04-07 14:00:09 -04:00 |
|
momentum_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
momentum_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
pooling_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
pooling_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
pooling_grad_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
pooling_grad_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
relu_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
relu_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
relu_grad_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
relu_grad_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
softmax_cross_entropy_with_logits_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
softmax_cross_entropy_with_logits_gpu_kernel.h
|
fix bug in cross entropy error
|
2020-04-21 14:21:46 +08:00 |
|
softmax_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
softmax_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
sparse_softmax_cross_entropy_with_logits_gpu_kernel.cc
|
initial version
|
2020-03-27 22:54:54 +08:00 |
|
sparse_softmax_cross_entropy_with_logits_gpu_kernel.h
|
initial version
|
2020-03-27 22:54:54 +08:00 |