Commit Graph

267 Commits (3f60bf6dfe1334c96747a845a3f4e1dcec29b6c6)

Author SHA1 Message Date
VectorSL b0a6346b54 gpu update cast type
5 years ago
mindspore-ci-bot fea930f7aa !4088 make gpu op Less to support int32
5 years ago
root 3b41023a6b add int32 cal for less gpu
5 years ago
mindspore-ci-bot 8f17535045 !3831 CUDA - GPU MirrorPad New Op
5 years ago
danish 081249b53f commit 1 - mirror pad
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
5 years ago
mindspore-ci-bot af9398b39a !3756 Change distribution api
5 years ago
yujianfeng 57cb1eeb14 Performance optimization of embedding_lookup
5 years ago
mindspore-ci-bot 49ba473bcc !3803 add gpu klDivLoss op
5 years ago
mindspore-ci-bot 51fcaf6e61 !3714 stridedslice/stridedslicegrad 4D to 7D
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
5 years ago
mindspore-ci-bot e1f93ec28c !3619 NMSWithMask - CUDA Impl
5 years ago
mindspore-ci-bot 82b103a740 !3780 add gpu BinaryCrossEntropy
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
5 years ago
mindspore-ci-bot 773f8e5352 !3707 Gpu support LayerNorm kernel
5 years ago
wilfChen 1eee3d6937 gpu layernorm
5 years ago
danish a2ffc9530e stuff added
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
5 years ago
wilfChen 9cad0fec1d gpu broadcast to
5 years ago
mindspore-ci-bot 46ae7879d4 !3394 Three distribution class instances: Exponential, Geometric Uniform distribution
5 years ago
mindspore-ci-bot 3cac1bb9f8 !3352 Add ROIAlignGrad GPU operator
5 years ago
mindspore-ci-bot efb12d5e1e !3560 Add random uniform real op at GPU end
5 years ago
mindspore-ci-bot bca16792be !3453 Support resolving an attribute of a Cell class instance
5 years ago
Zichun Ye 5b9899eb4f add support to resolve attr for a class
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
5 years ago
mindspore-ci-bot e730224a2c !3511 Add IOU, CheckValid, GatherNd, ScatterNd 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
5 years ago
baihuawei 4a8724e0ac add gpu cumsum
5 years ago
mindspore-ci-bot 183cf5cf5d !3285 Add Encode,Decode,SGD,floordiv,ScatterNd,GatherNd 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
5 years ago
mindspore-ci-bot 38a52a5b67 !3176 Add gpu support for ResizeNearestNeighbor
5 years ago
mindspore-ci-bot 5f10417b9f !3276 make gpu equal op support int32
5 years ago