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
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
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
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
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
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
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
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
mindspore-ci-bot
bca16792be
!3453 Support resolving an attribute of a Cell class instance
Merge pull request !3453 from zichun_ye/resolve_attr_pr
5 years ago
Zichun Ye
5b9899eb4f
add support to resolve attr for a class
update logic for cache
clean up code
add test cases for new api
update include of code
specify the resolve attr to ClassMember
avoid resolve attr of self
5 years ago
peixu_ren
e7a99397a5
Add random uniform real op at GPU end
5 years ago
peixu_ren
292f2de0cf
Add sampling functions in exponential, geometric and uniform distributions
5 years ago
Xun Deng
7bb83067d0
added exponential uniform geometric distribution
added cdf, logcdf, survival, logsurvival, entropy, mode, cross_entropy wrapper
5 years ago
mindspore-ci-bot
e730224a2c
!3511 Add IOU, CheckValid, GatherNd, ScatterNd ops.
Merge pull request !3511 from linqingke/new_ops
5 years ago
linqingke
7829bab811
add iou ops.
5 years ago
wilfChen
d3167da8af
gpu support stridedslice
5 years ago
tom__chen
5c3be0114f
add maxpool_with_argmax/grad cuda kernel
5 years ago
Jonathan Yan
ad40e00228
roi align grad v1
5 years ago
peixu_ren
19d80b87a9
Fix minor errors in probabilistic programming
5 years ago