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
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
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
mindspore-ci-bot
9c461f5565
!3513 Add parameter server mode_zoo case and CI test cases.
...
Merge pull request !3513 from ZPaC/master-add-ps-test-cases
5 years ago
ZPaC
07a52688c1
Add parameter server model_zoo case and CI test cases.
5 years ago
mindspore-ci-bot
ca42c21206
!3497 restore transformer st
...
Merge pull request !3497 from yuchaojie/transformer_st
5 years ago
linqingke
7829bab811
add iou ops.
5 years ago
mindspore-ci-bot
c22792aab1
!3495 Gpu StridedSlice performance optimize
...
Merge pull request !3495 from chenweifeng/strided_slice
5 years ago
wilfChen
d3167da8af
gpu support stridedslice
5 years ago
yuchaojie
a2b1c38cf2
restore transformer st
5 years ago
chenhaozhe
ef7ea3b92d
fix bert performance test case
5 years ago
mindspore-ci-bot
d21871a4fe
!3375 add bert ci script
...
Merge pull request !3375 from yoonlee666/bertci
5 years ago
mindspore-ci-bot
dc64a3878a
!3396 fix get dataset size error
...
Merge pull request !3396 from panfengfeng/fix_getdataset_size_error
5 years ago
tom__chen
5c3be0114f
add maxpool_with_argmax/grad cuda kernel
5 years ago
yoonlee666
4c167e3085
add bert ci script
5 years ago
panfengfeng
939e612906
fix get daataset size error
5 years ago
Jonathan Yan
ad40e00228
roi align grad v1
5 years ago