Xun Deng
be6db4a6fe
add scalar affine and softplus bijector
5 years ago
hanhuifeng2020
ab6f7420b5
modify some bug and add test case for gpu dropout op
5 years ago
peixu_ren
3820533ad1
Refactor Gamma and Poisson ops
5 years ago
VectorSL
b0a6346b54
gpu update cast type
5 years ago
changzherui
22336c0843
sync 0807 code to ms-incubator
5 years ago
linqingke
fb405ee6f4
broadcast, slice, scatter_nd ops optimizer.
5 years ago
Peilin Wang
3bd64bf03d
add unit test for resizeNearestNeighbor gpu op
fix pylint
fix pylint
changed mstype to np type to fix pylint
fix pylint
fix pylint
fix pylint
added back copyright
5 years ago
mindspore-ci-bot
fea930f7aa
!4088 make gpu op Less to support int32
Merge pull request !4088 from TuDouNi/master
5 years ago
panyifeng
34e50e5d6e
fix cell bprop
5 years ago
root
3b41023a6b
add int32 cal for less gpu
5 years ago
mindspore-ci-bot
8f17535045
!3831 CUDA - GPU MirrorPad New Op
Merge pull request !3831 from danishnxt/GPU_One
5 years ago
Wei Luning
d4d6457ea7
init parameter data by defaultOnly keep no data as MetaTensor in auto parallel mode
5 years ago
mindspore-ci-bot
1c6ddf70b6
!4043 add serving st
Merge pull request !4043 from hexia/serving_st
5 years ago
hexia
7f514a07e8
serving st
5 years ago
lvliang
1835ff33c4
optimize-the-time-of-producting-cacha-key-in-pynative
5 years ago
mindspore-ci-bot
6b14a1e85f
!3950 Add CI ps test cases.
Merge pull request !3950 from ZPaC/ci-add-ps-cases
5 years ago
danish
081249b53f
commit 1 - mirror pad
commit 2
lint fix
lint fix 2
updated backprop + st test
test_file_fix
test_file_fix_2
fixed header_guards
comments addressed
clangFormatFix
5 years ago
Jonathan Yan
43094bf78e
suport half for roi align
5 years ago
ZPaC
b10d4d6e0d
Add ps ci test cases.
5 years ago
peixu_ren
b9c8c0b439
Added Bijector TransformDistribution base classes and two instances: power and exp bijectors
5 years ago
mindspore-ci-bot
55d280e777
!3608 Add gpu support for RandomChoiceWithMask
Merge pull request !3608 from 34bunny/GPU-RandomChoiceWithMask
5 years ago
mindspore-ci-bot
af9398b39a
!3756 Change distribution api
Merge pull request !3756 from XunDeng/pp_poc_v3
5 years ago
yujianfeng
57cb1eeb14
Performance optimization of embedding_lookup
5 years ago
mindspore-ci-bot
be507b96e0
!3473 [refine]change base class of parameter
Merge pull request !3473 from vlne-v1/change_base_class_of_parameter
5 years ago
mindspore-ci-bot
49ba473bcc
!3803 add gpu klDivLoss op
Merge pull request !3803 from baihuawei/loss
5 years ago
mindspore-ci-bot
51fcaf6e61
!3714 stridedslice/stridedslicegrad 4D to 7D
Merge pull request !3714 from panbingao/stridedslice
5 years ago
Wei Luning
a05c38bb63
make python Parameter inherit from Tensor
5 years ago
TFbunny
ad8a786b07
add GPU support to RandomChoiceWithMask
5 years ago
Xun Deng
e87e1fc6bc
changed distribution api
5 years ago
mindspore-ci-bot
2449e4e7f1
!3760 Fix GPU-ResizeNearestNeighbor
Merge pull request !3760 from 34bunny/GPU-ResizeNearestNeighbor-fix
5 years ago
mindspore-ci-bot
e1f93ec28c
!3619 NMSWithMask - CUDA Impl
Merge pull request !3619 from danishnxt/GPU_One
5 years ago
mindspore-ci-bot
82b103a740
!3780 add gpu BinaryCrossEntropy
Merge pull request !3780 from baihuawei/losscuda
5 years ago
baihuawei
9eca56635d
add KLDiv loss
5 years ago
baihuawei
aa9ea1707c
add binary cross entropy
5 years ago
panbingao
e6335739d2
stridedslice/stridedslicegrad 4D to 7D
5 years ago
mindspore-ci-bot
31e889881b
!3737 Enlarge the threshold of resnet50 performance st in pynative
Merge pull request !3737 from JoyLvliang/enlarge-the-threshold-of-resnet50-performance-st-in-pynative
5 years ago
TFbunny
075acf80b1
fix ResizeNearestNeighbor & add util.cuh to support atomicadd for half
5 years ago
peixu_ren
374772035a
Refactor random uniform ops and complete gamma and poisson
5 years ago
wilfChen
4d600e70f1
gpu layernorm
5 years ago
lvliang
afe42e8d19
enlarge-the-threshold-of-resnet50-performance-st-in-pynative
5 years ago
mindspore-ci-bot
f1a39a0f72
!3655 gpu support BroadcastTo kernel
Merge pull request !3655 from chenweifeng/broadcast_to
5 years ago
mindspore-ci-bot
773f8e5352
!3707 Gpu support LayerNorm kernel
Merge pull request !3707 from chenweifeng/LayerNorm
5 years ago
wilfChen
1eee3d6937
gpu layernorm
5 years ago
danish
a2ffc9530e
stuff added
box copying fix
common function for IOU
new sort func
final update + unit test
remove one comment
fix lintig -1
lint fix 2
lint fix 3
last lint fix
value fix in ST nms with mask
addressing comments
pylint fix
pylint fix 1
test file fix
5 years ago
Xun Deng
e94d91ba95
remove import probability from nn/__init__.py
5 years ago
mindspore-ci-bot
afce1c3a40
!3341 GPU maxpool with argmax op
Merge pull request !3341 from tom_chen/maxpool_with_argmax
5 years ago
wilfChen
9cad0fec1d
gpu broadcast to
5 years ago
mindspore-ci-bot
46ae7879d4
!3394 Three distribution class instances: Exponential, Geometric Uniform distribution
Merge pull request !3394 from XunDeng/pp_poc_v3
5 years ago
mindspore-ci-bot
3cac1bb9f8
!3352 Add ROIAlignGrad GPU operator
Merge pull request !3352 from JonathanY/ROIAlignGrad
5 years ago
mindspore-ci-bot
efb12d5e1e
!3560 Add random uniform real op at GPU end
Merge pull request !3560 from peixu_ren/custom_gpu
5 years ago