| .. |
|
activation_gpu_kernel.cc
|
Add relu6 gpu
|
2020-09-14 21:48:02 +08:00 |
|
activation_gpu_kernel.h
|
change gpu kernel shape to size_t
|
2020-10-10 16:16:07 +08:00 |
|
activation_grad_kernel.cc
|
Add relu6 gpu
|
2020-09-14 21:48:02 +08:00 |
|
activation_grad_kernel.h
|
change gpu kernel shape to size_t
|
2020-10-10 16:16:07 +08:00 |
|
adam_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
adam_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
batchnorm_grad_gpu_kernel.cc
|
Add 2 reserve output for BatchNormGrad v1
|
2020-08-19 12:49:39 -04:00 |
|
batchnorm_grad_gpu_kernel.h
|
Add 2 reserve output for BatchNormGrad v1
|
2020-08-19 12:49:39 -04:00 |
|
bias_add_grad_gpu_kenel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
bias_add_grad_gpu_kenel.h
|
gpu inceptionv3 optimize
|
2020-09-04 14:45:34 +08:00 |
|
binary_cross_entropy_gpu_kernel.cc
|
add binary cross entropy
|
2020-07-31 10:45:08 +08:00 |
|
binary_cross_entropy_gpu_kernel.h
|
fix BCE and KLDiv
|
2020-09-28 10:26:15 +08:00 |
|
binary_cross_entropy_grad_kernel.cc
|
add binary cross entropy
|
2020-07-31 10:45:08 +08:00 |
|
binary_cross_entropy_grad_kernel.h
|
add binary cross entropy
|
2020-07-31 10:45:08 +08:00 |
|
combine_momentum_gpu_kernel.cc
|
gpu add combine mom fusion
|
2020-09-27 10:09:06 +08:00 |
|
combine_momentum_gpu_kernel.h
|
gpu add combine mom fusion
|
2020-09-27 10:09:06 +08:00 |
|
conv2d_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
conv2d_gpu_kernel.h
|
gpu fix codex
|
2020-08-24 19:57:05 +08:00 |
|
conv2d_grad_filter_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
conv2d_grad_filter_gpu_kernel.h
|
gpu fix codex
|
2020-08-24 19:57:05 +08:00 |
|
conv2d_grad_input_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
conv2d_grad_input_gpu_kernel.h
|
cudnn inplace optimizer
|
2020-09-30 16:18:16 +08:00 |
|
ctcloss_gpu_kernel.cc
|
add GPU CTCLoss
|
2020-09-22 08:24:17 +08:00 |
|
ctcloss_gpu_kernel.h
|
add GPU CTCLoss
|
2020-09-22 08:24:17 +08:00 |
|
dropout_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
dropout_gpu_kernel.h
|
modify some bug and add test case for gpu dropout op
|
2020-08-10 16:55:56 +08:00 |
|
dropout_grad_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
dropout_grad_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
flatten_gpu_kernel.cc
|
added type support for reshape gpu kernel
|
2020-08-20 13:52:23 -04:00 |
|
flatten_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
flatten_grad_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
flatten_grad_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
ftrl_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
ftrl_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
fused_adam_weight_decay.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
fused_adam_weight_decay.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
fused_batch_norm_ex_gpu_kernel.cc
|
add FusedBatchNoramEx gpu kernel
|
2020-08-19 20:51:32 +08:00 |
|
fused_batch_norm_ex_gpu_kernel.h
|
add FusedBatchNoramEx gpu kernel
|
2020-08-19 20:51:32 +08:00 |
|
fused_batch_norm_gpu_kernel.cc
|
gpu add fusion: replace momentum cast
|
2020-07-16 18:52:26 +08:00 |
|
fused_batch_norm_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
fused_batch_norm_grad_ex_gpu_kernel.cc
|
add FusedBatchNormGradEx gpu kernel
|
2020-08-20 10:59:40 +08:00 |
|
fused_batch_norm_grad_ex_gpu_kernel.h
|
cudnn inplace optimizer
|
2020-09-30 16:18:16 +08:00 |
|
fused_batchnorm_grad_gpu_kernel.cc
|
gpu add fusion: replace momentum cast
|
2020-07-16 18:52:26 +08:00 |
|
fused_batchnorm_grad_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
fused_scale_momentum_gpu_kernel.cc
|
gpu resnet50 fusion
|
2020-09-01 17:28:51 +08:00 |
|
fused_scale_momentum_gpu_kernel.h
|
gpu add combine mom fusion
|
2020-09-27 10:09:06 +08:00 |
|
fused_weightdecay_scale_momentum_gpu_kernel.cc
|
gpu resnet50 fusion
|
2020-09-01 17:28:51 +08:00 |
|
fused_weightdecay_scale_momentum_gpu_kernel.h
|
gpu add combine mom fusion
|
2020-09-27 10:09:06 +08:00 |
|
gelu_grad_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
gelu_grad_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
gelu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
gelu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
im2col_gpu_kernel.cc
|
support img2col for resnet50_thor GPU
|
2020-08-04 15:25:17 +08:00 |
|
im2col_gpu_kernel.h
|
support img2col for resnet50_thor GPU
|
2020-08-04 15:25:17 +08:00 |
|
kl_div_loss_gpu_kernel.cc
|
add KLDiv loss
|
2020-07-31 15:03:18 +08:00 |
|
kl_div_loss_gpu_kernel.h
|
fix BCE and KLDiv
|
2020-09-28 10:26:15 +08:00 |
|
kl_div_loss_grad_kernel.cc
|
add KLDiv loss
|
2020-07-31 15:03:18 +08:00 |
|
kl_div_loss_grad_kernel.h
|
add KLDiv loss
|
2020-07-31 15:03:18 +08:00 |
|
layer_norm_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
layer_norm_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
layer_norm_grad_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
layer_norm_grad_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
lstm_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
lstm_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
lstm_grad_data_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
lstm_grad_data_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
lstm_grad_weight_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
lstm_grad_weight_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
maxpool_with_argmax_gpu_kernel.cc
|
add maxpool_with_argmax/grad cuda kernel
|
2020-07-24 06:17:14 -08:00 |
|
maxpool_with_argmax_gpu_kernel.h
|
add maxpool_with_argmax/grad cuda kernel
|
2020-07-24 06:17:14 -08:00 |
|
maxpool_with_argmax_grad_gpu_kernel.cc
|
add maxpool_with_argmax/grad cuda kernel
|
2020-07-24 06:17:14 -08:00 |
|
maxpool_with_argmax_grad_gpu_kernel.h
|
add maxpool_with_argmax/grad cuda kernel
|
2020-07-24 06:17:14 -08:00 |
|
mirror_pad_gpu_kernel.cc
|
commit 1 - mirror pad
|
2020-08-05 16:48:58 -04:00 |
|
mirror_pad_gpu_kernel.h
|
code warning clean
|
2020-09-21 09:44:39 +08:00 |
|
mirror_pad_grad_gpu_kernel.cc
|
commit 1 - mirror pad
|
2020-08-05 16:48:58 -04:00 |
|
mirror_pad_grad_gpu_kernel.h
|
code warning clean
|
2020-09-21 09:44:39 +08:00 |
|
momentum_gpu_kernel.cc
|
gpu add fusion: replace momentum cast
|
2020-07-16 18:52:26 +08:00 |
|
momentum_gpu_kernel.h
|
GPU momentum supports use_nesterov
|
2020-09-19 19:24:19 +08:00 |
|
pad_gpu_kernel.cc
|
commit one - PadOp Support files
|
2020-07-22 13:41:11 -04:00 |
|
pad_gpu_kernel.h
|
New User facing Pad Kernel + ST + Allows for channel padding
|
2020-09-04 15:13:05 -04:00 |
|
pooling_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
pooling_gpu_kernel.h
|
fix bn cast
|
2020-09-19 10:25:44 +08:00 |
|
pooling_grad_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
pooling_grad_gpu_kernel.h
|
!6532 GPU fix BnCast
|
2020-09-19 15:58:10 +08:00 |
|
rmsprop_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
rmsprop_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
roi_align_gpu_kernel.cc
|
code warning clean
|
2020-09-21 09:44:39 +08:00 |
|
roi_align_gpu_kernel.h
|
roi align memory leak
|
2020-09-02 22:17:59 -04:00 |
|
roi_align_grad_gpu_kernel.cc
|
code warning clean
|
2020-09-21 09:44:39 +08:00 |
|
roi_align_grad_gpu_kernel.h
|
roi_align grad init
|
2020-09-15 09:23:38 -04:00 |
|
sgd_gpu_kernel.cc
|
gpu ops code and test case.
|
2020-07-22 20:15:52 +08:00 |
|
sgd_gpu_kernel.h
|
gpu ops code and test case.
|
2020-07-22 20:15:52 +08:00 |
|
sigmoid_cross_entropy_with_logits_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
sigmoid_cross_entropy_with_logits_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
sigmoid_cross_entropy_with_logits_grad_gpu_kernel.h
|
bugfix:SigmoidCrossEntropyWithLogitsGrad need multiply dout
|
2020-08-17 17:00:43 +08:00 |
|
smooth_l1_loss_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
smooth_l1_loss_gpu_kernel.h
|
modified documentation and gpu kernel for smoothL1Loss
|
2020-08-19 17:06:22 -04:00 |
|
smooth_l1_loss_grad_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
smooth_l1_loss_grad_gpu_kernel.h
|
modified documentation and gpu kernel for smoothL1Loss
|
2020-08-19 17:06:22 -04:00 |
|
softmax_cross_entropy_with_logits_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
softmax_cross_entropy_with_logits_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |
|
softmax_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
softmax_gpu_kernel.h
|
change gpu kernel shape to size_t
|
2020-10-10 16:16:07 +08:00 |
|
softmax_grad_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
softmax_grad_gpu_kernel.h
|
change gpu kernel shape to size_t
|
2020-10-10 16:16:07 +08:00 |
|
sparse_softmax_cross_entropy_with_logits_gpu_kernel.cc
|
mindspore path adjust
|
2020-07-14 18:07:28 +08:00 |
|
sparse_softmax_cross_entropy_with_logits_gpu_kernel.h
|
Unified code style
|
2020-07-20 10:51:55 +08:00 |