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.
29867 Commits
18 Branches
584 MB
67 Download
 
 
 
 
 
 
Tree: 062dcaaaa1
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '062dcaaaa1'
${ noResults }
mindspore2022/mindspore/ccsrc/backend/kernel_compiler/gpu/nn
djc b077aa1cab [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
..
activation_gpu_kernel.cc add dtype supports for relu/reluv2/relugradv2 5 years ago
activation_gpu_kernel.h update tensor size check func 4 years ago
activation_grad_kernel.cc add relugrad gpu 5 years ago
activation_grad_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
adagrad_gpu_kernel.cc I3AP06: dtype and return value 5 years ago
adagrad_gpu_kernel.h codex 4 years ago
adam_gpu_kernel.cc mindspore path adjust 5 years ago
adam_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
adam_weight_decay_gpu_kernel.cc add AdamWeightDecayOp 5 years ago
adam_weight_decay_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
adaptive_avg_pool2d_gpu_kernel.cc Add ops test PR 1 4 years ago
adaptive_avg_pool2d_gpu_kernel.h initial commit: fix 11 dts tickets 4 years ago
apply_gradient_descent_gpu_kernel.cc add gpu ApplyGradientDescent 4 years ago
apply_gradient_descent_gpu_kernel.h add gpu ApplyGradientDescent 4 years ago
batch_norm_gpu_kernel.cc IR operators of GPU and CPU are unified as batchnorm 5 years ago
batch_norm_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
batch_norm_grad_gpu_kernel.cc IR operators of GPU and CPU are unified as batchnorm 5 years ago
batch_norm_grad_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
bce_with_logits_loss_kernel.cc add gpu BCEWithLogitsLoss kernel 4 years ago
bce_with_logits_loss_kernel.h add gpu BCEWithLogitsLoss kernel 4 years ago
bias_add_gpu_kernel.cc add dynamic shape and testcases to GPU biasadd 5 years ago
bias_add_gpu_kernel.h !11247 Add dynamic shape support to GPU BiasAdd 5 years ago
bias_add_grad_gpu_kenel.cc 1.Optimize bias add grad kernel 4 years ago
bias_add_grad_gpu_kenel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
binary_cross_entropy_gpu_kernel.cc binary_cross_entropy op supports optional weight input 4 years ago
binary_cross_entropy_gpu_kernel.h binary_cross_entropy op supports optional weight input 4 years ago
binary_cross_entropy_grad_kernel.cc binary_cross_entropy op supports optional weight input 4 years ago
binary_cross_entropy_grad_kernel.h binary_cross_entropy op supports optional weight input 4 years ago
combine_momentum_gpu_kernel.cc gpu add combine mom fusion 5 years ago
combine_momentum_gpu_kernel.h gpu nhwc 5 years ago
conv2d_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
conv2d_grad_filter_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_grad_filter_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
conv2d_grad_input_gpu_kernel.cc mindspore path adjust 5 years ago
conv2d_grad_input_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
conv3d_gpu_kernel.cc conv3d gpu op 4 years ago
conv3d_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
conv3d_grad_filter_gpu_kernel.cc conv3d grad gpu op 4 years ago
conv3d_grad_filter_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
conv3d_grad_input_gpu_kernel.cc conv3d grad gpu op 4 years ago
conv3d_grad_input_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
conv3d_transpose_gpu_kernel.cc Support Conv3dTranspose for GPU 4 years ago
conv3d_transpose_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
ctcloss_gpu_kernel.cc add GPU CTCLoss 5 years ago
ctcloss_gpu_kernel.h cpu support scalar tensor 5 years ago
dropout3d_gpu_kernel.cc implemented new gpu op, Dropout3D 4 years ago
dropout3d_gpu_kernel.h implemented new gpu op, Dropout3D 4 years ago
dropout_gpu_kernel.cc mindspore path adjust 5 years ago
dropout_gpu_kernel.h fix dropout and matrix_inverse op bug. 4 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 add dtypes for sponge 4 years ago
flatten_gpu_kernel.h gpu suppor dynamic shape 5 years ago
flatten_grad_gpu_kernel.cc mindspore path adjust 5 years ago
flatten_grad_gpu_kernel.h gpu suppor dynamic shape 5 years ago
ftrl_gpu_kernel.cc mindspore path adjust 5 years ago
ftrl_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
fused_adam_weight_decay.cc mindspore path adjust 5 years ago
fused_adam_weight_decay.h fix gpu adam fusion kernel 5 years ago
fused_add_relu_grad_v2_gpu_kernel.cc relu optimize 5 years ago
fused_add_relu_grad_v2_gpu_kernel.h relu optimize 5 years ago
fused_add_relu_v2_gpu_kernel.cc relu optimize 5 years ago
fused_add_relu_v2_gpu_kernel.h relu optimize 5 years ago
fused_scale_momentum_gpu_kernel.cc gpu resnet50 fusion 5 years ago
fused_scale_momentum_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
fused_weightdecay_momentum_gpu_kernel.cc momentum weightdecay fusion 5 years ago
fused_weightdecay_momentum_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
fused_weightdecay_scale_momentum_gpu_kernel.cc gpu resnet50 fusion 5 years ago
fused_weightdecay_scale_momentum_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
gelu_grad_kernel.cc modify Gelu、FastGelu to GeLU and FastGeLU 5 years ago
gelu_grad_kernel.h Unified code style 5 years ago
gelu_kernel.cc modify Gelu、FastGelu to GeLU and FastGeLU 5 years ago
gelu_kernel.h Unified code style 5 years ago
hsigmoid_gpu_kernel.cc add gpu kernel HSigmoid and HSigmoidGrad 4 years ago
hsigmoid_gpu_kernel.h add gpu kernel HSigmoid and HSigmoidGrad 4 years ago
hsigmoid_grad_gpu_kernel.cc add gpu kernel HSigmoid and HSigmoidGrad 4 years ago
hsigmoid_grad_gpu_kernel.h add fix to hsigmoid grad 4 years ago
hswish_gpu_kernel.cc develop hswish and hswish grad gpu operators 4 years ago
hswish_gpu_kernel.h develop hswish and hswish grad gpu operators 4 years ago
hswish_grad_gpu_kernel.cc develop hswish and hswish grad gpu operators 4 years ago
hswish_grad_gpu_kernel.h code security check 4 years ago
im2col_gpu_kernel.cc support img2col for resnet50_thor GPU 5 years ago
im2col_gpu_kernel.h 05311 sponge_ops update_thor_gradient thor_ops.py im2col 4 years ago
instance_norm_gpu_kernel.cc add new op instancenorm2d 5 years ago
instance_norm_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
instance_norm_grad_gpu_kernel.cc add new op instancenorm2d 5 years ago
instance_norm_grad_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
kl_div_loss_gpu_kernel.cc fixBCE 5 years ago
kl_div_loss_gpu_kernel.h fix BCE and KLDiv 5 years ago
kl_div_loss_grad_kernel.cc fixBCE 5 years ago
kl_div_loss_grad_kernel.h add KLDiv loss 5 years ago
l2_loss_gpu_kernel.cc add support to op L2Loss on gpu 5 years ago
l2_loss_gpu_kernel.h add support to op L2Loss on gpu 5 years ago
l2normalize_gpu_kernel.cc add l2normalize gpu kernel. 5 years ago
l2normalize_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
l2normalize_grad_gpu_kernel.cc new add l2normalize_grad gpu ops. 5 years ago
l2normalize_grad_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
layer_norm_gpu_kernel.cc mindspore path adjust 5 years ago
layer_norm_gpu_kernel.h support int64 shape 5 years ago
layer_norm_grad_gpu_kernel.cc mindspore path adjust 5 years ago
layer_norm_grad_gpu_kernel.h support int64 shape 5 years ago
layer_norm_grad_grad_gpu_kernel.cc Add new operator layer_norm_grad_grad 5 years ago
layer_norm_grad_grad_gpu_kernel.h ci format problem fix. 4 years ago
local_response_norm_gpu_kernel.cc Local Reponse Norm GPU Op 4 years ago
local_response_norm_gpu_kernel.h Local Reponse Norm GPU Op 4 years ago
local_response_norm_grad_gpu_kernel.cc Local Reponse Norm GPU Op 4 years ago
local_response_norm_grad_gpu_kernel.h Local Reponse Norm GPU Op 4 years ago
lstm_gpu_kernel.cc mindspore path adjust 5 years ago
lstm_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
lstm_grad_data_gpu_kernel.cc mindspore path adjust 5 years ago
lstm_grad_data_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
lstm_grad_weight_gpu_kernel.cc mindspore path adjust 5 years ago
lstm_grad_weight_gpu_kernel.h fix lstm gpu 4 years ago
maxpool_with_argmax_gpu_kernel.cc add maxpool_with_argmax/grad cuda kernel 5 years ago
maxpool_with_argmax_gpu_kernel.h initial commit: fix 11 dts tickets 4 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 rewrite gpu implementation of op maxpool_with_argmax and maxpool_with_argmax_grad 5 years ago
mirror_pad_gpu_kernel.cc Reworked mirrorPad 5 years ago
mirror_pad_gpu_kernel.h Reworked mirrorPad 5 years ago
mirror_pad_grad_gpu_kernel.cc Reworked mirrorPad 5 years ago
mirror_pad_grad_gpu_kernel.h Reworked mirrorPad 5 years ago
momentum_gpu_kernel.cc gpu add fusion: replace momentum cast 5 years ago
momentum_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
nll_loss_gpu_kernel.cc implemented nll_loss operation for gpu 4 years ago
nll_loss_gpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
nll_loss_grad_gpu_kernel.cc added nll_loss_grad for gpu 4 years ago
nll_loss_grad_gpu_kernel.h added nll_loss_grad for gpu 4 years ago
pad_gpu_kernel.cc Reworked mirrorPad 5 years ago
pad_gpu_kernel.h pad_CI_Alarm fix 5 years ago
pooling_gpu_kernel.cc support MaxPool3D(CPU/GPU), AvgPool3D(CPU/GPU) and Conv3D(CPU) 4 years ago
pooling_gpu_kernel.h support MaxPool3DGrad on GPU 4 years ago
pooling_grad_gpu_kernel.cc support MaxPool3DGrad on GPU 4 years ago
pooling_grad_gpu_kernel.h support MaxPool3DGrad on GPU 4 years ago
prelu_gpu_kernel.cc support PReLU for GPU platform 4 years ago
prelu_gpu_kernel.h improve PReLU forward and implement backward on GPU 4 years ago
prelu_grad_gpu_kernel.cc improve PReLU forward and implement backward on GPU 4 years ago
prelu_grad_gpu_kernel.h fix prelu weight grad accuracy error fp16 on GPU 4 years ago
relu_gpu_kernel.cc add unsigned type support to relu and relugrad gpu 5 years ago
relu_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
relu_grad_gpu_kernel.cc add unsigned type support to relu and relugrad gpu 5 years ago
relu_grad_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
relu_grad_v2_gpu_kernel.cc add unsigned type support to relu and relugrad gpu 5 years ago
relu_grad_v2_gpu_kernel.h relu optimize 5 years ago
relu_v2_gpu_kernel.cc add unsigned type support to relu and relugrad gpu 5 years ago
relu_v2_gpu_kernel.h add dtype supports for relu/reluv2/relugradv2 5 years ago
resize_bilinear_gpu_kernel.cc resize_bilinear gpu op 4 years ago
resize_bilinear_gpu_kernel.h resize_bilinear gpu op 4 years ago
resize_bilinear_grad_gpu_kernel.cc add resize bilinear grad gpu op 4 years ago
resize_bilinear_grad_gpu_kernel.h fix for async mem_init bilinearResize_grad 4 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 support int64 shape 5 years ago
roi_align_grad_gpu_kernel.cc code warning clean 5 years ago
roi_align_grad_gpu_kernel.h support int64 shape 5 years ago
sgd_gpu_kernel.cc gpu ops code and test case. 5 years ago
sgd_gpu_kernel.h fix gpu ops bug. 5 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.cc add float64 support to SigmoidCrossEntropyWithLogits and Grad 5 years ago
sigmoid_cross_entropy_with_logits_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.cc add float64 support to SigmoidCrossEntropyWithLogits and Grad 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 initial commit: fix 11 dts tickets 4 years ago
softmax_gpu_kernel.cc mindspore path adjust 5 years ago
softmax_gpu_kernel.h add softmax on specified axis 4 years ago
softmax_grad_gpu_kernel.cc mindspore path adjust 5 years ago
softmax_grad_gpu_kernel.h add trace for gpu error/excpt log 5 years ago
softplus_gpu_kernel.cc new add softplus and softplus grad gpu ops. 5 years ago
softplus_gpu_kernel.h new add softplus and softplus grad gpu ops. 5 years ago
softplus_grad_gpu_kernel.cc new add softplus and softplus grad gpu ops. 5 years ago
softplus_grad_gpu_kernel.h new add softplus and softplus grad gpu ops. 5 years ago
sparse_apply_proximal_adagrad_kernel.cc add GPU SparseApplyProximalAdagrad 5 years ago
sparse_apply_proximal_adagrad_kernel.h fix the exception when occur error and replace magic number with const value. 4 years ago
sparse_ftrl_gpu_kernel.cc first commit - updated op + STs 5 years ago
sparse_ftrl_gpu_kernel.h fix the exception when occur error and replace magic number with const value. 4 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 add trace for gpu error/excpt log 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