zzy34407230
/
mindspore2022
Not watched
1
0
Fork 0
Code
Releases 22 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
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.
11477 Commits
18 Branches
584 MB
67 Download
 
 
 
 
 
 
Tree: bada826b18
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'bada826b18'
${ noResults }
mindspore2022/mindspore/ccsrc/backend/kernel_compiler/gpu/nn
mindspore-ci-bot 483f1aca9d !7106 GPU change kernel shape to size_t 5 years ago
..
activation_gpu_kernel.cc Add relu6 gpu 5 years ago
activation_gpu_kernel.h change gpu kernel shape to size_t 5 years ago
activation_grad_kernel.cc Add relu6 gpu 5 years ago
activation_grad_kernel.h change gpu kernel shape to size_t 5 years ago
adam_gpu_kernel.cc mindspore path adjust 5 years ago
adam_gpu_kernel.h Unified code style 5 years ago
batchnorm_grad_gpu_kernel.cc Add 2 reserve output for BatchNormGrad v1 5 years ago
batchnorm_grad_gpu_kernel.h Add 2 reserve output for BatchNormGrad v1 5 years ago
bias_add_grad_gpu_kenel.cc mindspore path adjust 5 years ago
bias_add_grad_gpu_kenel.h gpu inceptionv3 optimize 5 years ago
binary_cross_entropy_gpu_kernel.cc add binary cross entropy 5 years ago
binary_cross_entropy_gpu_kernel.h fix BCE and KLDiv 5 years ago
binary_cross_entropy_grad_kernel.cc add binary cross entropy 5 years ago
binary_cross_entropy_grad_kernel.h add binary cross entropy 5 years ago
combine_momentum_gpu_kernel.cc gpu add combine mom fusion 5 years ago
combine_momentum_gpu_kernel.h gpu add combine mom fusion 5 years ago
conv2d_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_gpu_kernel.h gpu fix codex 5 years ago
conv2d_grad_filter_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_grad_filter_gpu_kernel.h gpu fix codex 5 years ago
conv2d_grad_input_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_grad_input_gpu_kernel.h cudnn inplace optimizer 5 years ago
ctcloss_gpu_kernel.cc add GPU CTCLoss 5 years ago
ctcloss_gpu_kernel.h add GPU CTCLoss 5 years ago
dropout_gpu_kernel.cc mindspore path adjust 5 years ago
dropout_gpu_kernel.h modify some bug and add test case for gpu dropout op 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 added type support for reshape gpu kernel 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_ex_gpu_kernel.cc add FusedBatchNoramEx gpu kernel 5 years ago
fused_batch_norm_ex_gpu_kernel.h add FusedBatchNoramEx gpu kernel 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_batch_norm_grad_ex_gpu_kernel.cc add FusedBatchNormGradEx gpu kernel 5 years ago
fused_batch_norm_grad_ex_gpu_kernel.h cudnn inplace optimizer 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
fused_scale_momentum_gpu_kernel.cc gpu resnet50 fusion 5 years ago
fused_scale_momentum_gpu_kernel.h gpu add combine mom fusion 5 years ago
fused_weightdecay_scale_momentum_gpu_kernel.cc gpu resnet50 fusion 5 years ago
fused_weightdecay_scale_momentum_gpu_kernel.h gpu add combine mom fusion 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
im2col_gpu_kernel.cc support img2col for resnet50_thor GPU 5 years ago
im2col_gpu_kernel.h support img2col for resnet50_thor GPU 5 years ago
kl_div_loss_gpu_kernel.cc add KLDiv loss 5 years ago
kl_div_loss_gpu_kernel.h fix BCE and KLDiv 5 years ago
kl_div_loss_grad_kernel.cc add KLDiv loss 5 years ago
kl_div_loss_grad_kernel.h add KLDiv loss 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
mirror_pad_gpu_kernel.cc commit 1 - mirror pad 5 years ago
mirror_pad_gpu_kernel.h code warning clean 5 years ago
mirror_pad_grad_gpu_kernel.cc commit 1 - mirror pad 5 years ago
mirror_pad_grad_gpu_kernel.h code warning clean 5 years ago
momentum_gpu_kernel.cc gpu add fusion: replace momentum cast 5 years ago
momentum_gpu_kernel.h GPU momentum supports use_nesterov 5 years ago
pad_gpu_kernel.cc commit one - PadOp Support files 5 years ago
pad_gpu_kernel.h New User facing Pad Kernel + ST + Allows for channel padding 5 years ago
pooling_gpu_kernel.cc mindspore path adjust 5 years ago
pooling_gpu_kernel.h fix bn cast 5 years ago
pooling_grad_gpu_kernel.cc mindspore path adjust 5 years ago
pooling_grad_gpu_kernel.h !6532 GPU fix BnCast 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 code warning clean 5 years ago
roi_align_gpu_kernel.h roi align memory leak 5 years ago
roi_align_grad_gpu_kernel.cc code warning clean 5 years ago
roi_align_grad_gpu_kernel.h roi_align grad init 5 years ago
sgd_gpu_kernel.cc gpu ops code and test case. 5 years ago
sgd_gpu_kernel.h gpu ops code and test case. 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 bugfix:SigmoidCrossEntropyWithLogitsGrad need multiply dout 5 years ago
smooth_l1_loss_gpu_kernel.cc mindspore path adjust 5 years ago
smooth_l1_loss_gpu_kernel.h modified documentation and gpu kernel for smoothL1Loss 5 years ago
smooth_l1_loss_grad_gpu_kernel.cc mindspore path adjust 5 years ago
smooth_l1_loss_grad_gpu_kernel.h modified documentation and gpu kernel for smoothL1Loss 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 change gpu kernel shape to size_t 5 years ago
softmax_grad_gpu_kernel.cc mindspore path adjust 5 years ago
softmax_grad_gpu_kernel.h change gpu kernel shape to size_t 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
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

阿对对队

C++ Python Text C Unity3D Asset other