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.
 
 
 
 
 
 
panfengfeng ca881ec03e add maxpool_with_argmax/grad cuda kernel 5 years ago
..
activation_gpu_kernel.cc mindspore path adjust 5 years ago
activation_gpu_kernel.h Unified code style 5 years ago
activation_grad_kernel.cc mindspore path adjust 5 years ago
activation_grad_kernel.h Unified code style 5 years ago
adam_gpu_kernel.cc mindspore path adjust 5 years ago
adam_gpu_kernel.h Unified code style 5 years ago
bias_add_grad_gpu_kenel.cc mindspore path adjust 5 years ago
bias_add_grad_gpu_kenel.h Unified code style 5 years ago
conv2d_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_gpu_kernel.h add int64-->fp16 and update conv pad 5 years ago
conv2d_grad_filter_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_grad_filter_gpu_kernel.h add int64-->fp16 and update conv pad 5 years ago
conv2d_grad_input_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_grad_input_gpu_kernel.h add int64-->fp16 and update conv pad 5 years ago
ctcloss_gpu_kernel.cc mindspore path adjust 5 years ago
ctcloss_gpu_kernel.h !3244 Support 2-dimensions target of CTCLossV2 5 years ago
dropout_gpu_kernel.cc mindspore path adjust 5 years ago
dropout_gpu_kernel.h Unified code style 5 years ago
dropout_grad_kernel.cc mindspore path adjust 5 years ago
dropout_grad_kernel.h Unified code style 5 years ago
flatten_gpu_kernel.cc mindspore path adjust 5 years ago
flatten_gpu_kernel.h Unified code style 5 years ago
flatten_grad_gpu_kernel.cc mindspore path adjust 5 years ago
flatten_grad_gpu_kernel.h Unified code style 5 years ago
ftrl_gpu_kernel.cc mindspore path adjust 5 years ago
ftrl_gpu_kernel.h Unified code style 5 years ago
fused_adam_weight_decay.cc mindspore path adjust 5 years ago
fused_adam_weight_decay.h Unified code style 5 years ago
fused_batch_norm_gpu_kernel.cc gpu add fusion: replace momentum cast 5 years ago
fused_batch_norm_gpu_kernel.h Unified code style 5 years ago
fused_batchnorm_grad_gpu_kernel.cc gpu add fusion: replace momentum cast 5 years ago
fused_batchnorm_grad_gpu_kernel.h Unified code style 5 years ago
gelu_grad_kernel.cc mindspore path adjust 5 years ago
gelu_grad_kernel.h Unified code style 5 years ago
gelu_kernel.cc mindspore path adjust 5 years ago
gelu_kernel.h Unified code style 5 years ago
layer_norm_gpu_kernel.cc mindspore path adjust 5 years ago
layer_norm_gpu_kernel.h Unified code style 5 years ago
layer_norm_grad_gpu_kernel.cc mindspore path adjust 5 years ago
layer_norm_grad_gpu_kernel.h Unified code style 5 years ago
lstm_gpu_kernel.cc mindspore path adjust 5 years ago
lstm_gpu_kernel.h Unified code style 5 years ago
lstm_grad_data_gpu_kernel.cc mindspore path adjust 5 years ago
lstm_grad_data_gpu_kernel.h Unified code style 5 years ago
lstm_grad_weight_gpu_kernel.cc mindspore path adjust 5 years ago
lstm_grad_weight_gpu_kernel.h Unified code style 5 years ago
maxpool_with_argmax_gpu_kernel.cc add maxpool_with_argmax/grad cuda kernel 5 years ago
maxpool_with_argmax_gpu_kernel.h add maxpool_with_argmax/grad cuda kernel 5 years ago
maxpool_with_argmax_grad_gpu_kernel.cc add maxpool_with_argmax/grad cuda kernel 5 years ago
maxpool_with_argmax_grad_gpu_kernel.h add maxpool_with_argmax/grad cuda kernel 5 years ago
momentum_gpu_kernel.cc gpu add fusion: replace momentum cast 5 years ago
momentum_gpu_kernel.h Unified code style 5 years ago
pooling_gpu_kernel.cc mindspore path adjust 5 years ago
pooling_gpu_kernel.h Unified code style 5 years ago
pooling_grad_gpu_kernel.cc mindspore path adjust 5 years ago
pooling_grad_gpu_kernel.h Unified code style 5 years ago
rmsprop_gpu_kernel.cc mindspore path adjust 5 years ago
rmsprop_gpu_kernel.h Unified code style 5 years ago
roi_align_gpu_kernel.cc roi align v1 5 years ago
roi_align_gpu_kernel.h roi align v1 5 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.cc mindspore path adjust 5 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.h Unified code style 5 years ago
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.cc mindspore path adjust 5 years ago
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.h Unified code style 5 years ago
smooth_l1_loss_gpu_kernel.cc mindspore path adjust 5 years ago
smooth_l1_loss_gpu_kernel.h Unified code style 5 years ago
smooth_l1_loss_grad_gpu_kernel.cc mindspore path adjust 5 years ago
smooth_l1_loss_grad_gpu_kernel.h Unified code style 5 years ago
softmax_cross_entropy_with_logits_gpu_kernel.cc mindspore path adjust 5 years ago
softmax_cross_entropy_with_logits_gpu_kernel.h Unified code style 5 years ago
softmax_gpu_kernel.cc mindspore path adjust 5 years ago
softmax_gpu_kernel.h Unified code style 5 years ago
softmax_grad_gpu_kernel.cc mindspore path adjust 5 years ago
softmax_grad_gpu_kernel.h Unified code style 5 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.cc mindspore path adjust 5 years ago
sparse_softmax_cross_entropy_with_logits_gpu_kernel.h Unified code style 5 years ago