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
mindspore-ci-bot
55d280e777
!3608 Add gpu support for RandomChoiceWithMask
Merge pull request !3608 from 34bunny/GPU-RandomChoiceWithMask
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
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
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
3cac1bb9f8
!3352 Add ROIAlignGrad GPU operator
Merge pull request !3352 from JonathanY/ROIAlignGrad
5 years ago
peixu_ren
e7a99397a5
Add random uniform real op at GPU end
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
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
Jonathan Yan
661b993475
roi align v1
5 years ago
TFbunny
de28cdc71c
add GPU support to ResizeNearestNeighbor
5 years ago
mindspore-ci-bot
cf4353f728
!3220 Add random normal op at MindSpore front-end
Merge pull request !3220 from peixu_ren/custom_gpu
5 years ago
qujianwei
7479fb24a0
make gpu equal op support int32
5 years ago
peixu_ren
9b45018dfd
Add random normal op at MindSpore front-end
5 years ago
VectorSL
90f15df037
add int64-->fp16 and update conv pad
5 years ago
mindspore-ci-bot
32921ea3dc
!3166 add gpu oneslike op
Merge pull request !3166 from qujianwei/gpu-oneslike
5 years ago
qujianwei
fb2ac74d9a
add gpu oneslike kernel
5 years ago
mindspore-ci-bot
11732f0ea2
!3135 GPU cast support more type
Merge pull request !3135 from VectorSL/cast2
5 years ago
VectorSL
aef2c1984e
cast support more types
5 years ago
mindspore-ci-bot
251683096a
!3045 Gpu support TopK kernel
Merge pull request !3045 from chenweifeng/sort
5 years ago
mindspore-ci-bot
e249197c73
!3003 gpu support BroadcastTo kernels
Merge pull request !3003 from chenweifeng/broadcastto
5 years ago
mindspore-ci-bot
ad09bf3e87
!3083 add gpu split and restructure gpu concat
Merge pull request !3083 from zhaoting/master
5 years ago