You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
caifubi 5b963aef2b Change uintptr_t to void ptr 5 years ago
..
bias_add_grad_gpu_kenel.cc initial version 6 years ago
bias_add_grad_gpu_kenel.h Change uintptr_t to void ptr 5 years ago
conv2d_gpu_kernel.cc initial version 6 years ago
conv2d_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
conv2d_grad_filter_gpu_kernel.cc initial version 6 years ago
conv2d_grad_filter_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
conv2d_grad_input_gpu_kernel.cc initial version 6 years ago
conv2d_grad_input_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
dropout_gpu_kernel.cc Change uintptr_t to void ptr 5 years ago
dropout_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
dropout_grad_kernel.cc Change uintptr_t to void ptr 5 years ago
dropout_grad_kernel.h Change uintptr_t to void ptr 5 years ago
flatten_gpu_kernel.cc initial version 6 years ago
flatten_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
flatten_grad_gpu_kernel.cc initial version 6 years ago
flatten_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
fused_batch_norm_gpu_kernel.cc modify ResizeNearestNeighborV2D 5 years ago
fused_batch_norm_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
fused_batchnorm_grad_gpu_kernel.cc initial version 6 years ago
fused_batchnorm_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
gelu_grad_kernel.cc gpu support Gelu & GeluGrad kernels 5 years ago
gelu_grad_kernel.h Change uintptr_t to void ptr 5 years ago
gelu_kernel.cc gpu support Gelu & GeluGrad kernels 5 years ago
gelu_kernel.h Change uintptr_t to void ptr 5 years ago
layer_norm_gpu_kernel.cc Gpu support LayerNorm kernel 5 years ago
layer_norm_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
layer_norm_grad_gpu_kernel.cc Gpu support LayerNorm kernel 5 years ago
layer_norm_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
lstm_gpu_kernel.cc initial version 6 years ago
lstm_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
lstm_grad_data_gpu_kernel.cc initial version 6 years ago
lstm_grad_data_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
lstm_grad_weight_gpu_kernel.cc initial version 6 years ago
lstm_grad_weight_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
momentum_gpu_kernel.cc initial version 6 years ago
momentum_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
pooling_gpu_kernel.cc initial version 6 years ago
pooling_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
pooling_grad_gpu_kernel.cc initial version 6 years ago
pooling_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
relu_gpu_kernel.cc initial version 6 years ago
relu_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
relu_grad_kernel.cc initial version 6 years ago
relu_grad_kernel.h Change uintptr_t to void ptr 5 years ago
rmsprop_gpu_kernel.cc LayerNormGrad fix & codex 5 years ago
rmsprop_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
softmax_cross_entropy_with_logits_gpu_kernel.cc initial version 6 years ago
softmax_cross_entropy_with_logits_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
softmax_gpu_kernel.cc gpu support logsoftmax & logsoftmaxgrad kernel 5 years ago
softmax_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
softmax_grad_gpu_kernel.cc gpu support logsoftmax & logsoftmaxgrad kernel 5 years ago
softmax_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.cc initial version 6 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
tanh_gpu_kernel.cc gpu support tanh & tanhgrad kernel 5 years ago
tanh_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
tanh_grad_kernel.cc gpu support tanh & tanhgrad kernel 5 years ago
tanh_grad_kernel.h Change uintptr_t to void ptr 5 years ago