VectorSL
b0a6346b54
gpu update cast type
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
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
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
mindspore-ci-bot
1e88d64b20
!3259 add gpu cumsum op and reducemin
...
Merge pull request !3259 from baihuawei/automl
5 years ago
baihuawei
4a8724e0ac
add gpu cumsum
5 years ago
mindspore-ci-bot
183cf5cf5d
!3285 Add Encode,Decode,SGD,floordiv,ScatterNd,GatherNd ops.
...
Merge pull request !3285 from linqingke/gpu_ops
5 years ago
linqingke
f679568d86
gpu ops code and test case.
5 years ago
kingfo
09b437285b
support mix precision in pynative
5 years ago
mindspore-ci-bot
d15b4c5d61
!3201 RoI Align GPU kernel
...
Merge pull request !3201 from JonathanY/main
5 years ago
mindspore-ci-bot
38a52a5b67
!3176 Add gpu support for ResizeNearestNeighbor
...
Merge pull request !3176 from 34bunny/GPU-ResizeNearestNeighbor
5 years ago
mindspore-ci-bot
5f10417b9f
!3276 make gpu equal op support int32
...
Merge pull request !3276 from qujianwei/master
5 years ago