i-robot
|
2c9fe2bc6b
|
!34869 support tensor to sparse tensor conversion
Merge pull request !34869 from 杨林枫/dense_to_csr_gpu
|
2022-06-11 07:03:19 +00:00 |
i-robot
|
f50df36ab7
|
!35738 fix gpu KLDivLoss st
Merge pull request !35738 from baihuawei/ci
|
2022-06-11 02:42:26 +00:00 |
i-robot
|
b40c8385b9
|
!35685 update InstanceNorm GPU kernel
Merge pull request !35685 from zhujingxuan/InstanceNorm
|
2022-06-10 10:16:47 +00:00 |
i-robot
|
402431e754
|
!35667 [kernel]add celu vmap
Merge pull request !35667 from 张学同/celu
|
2022-06-10 08:21:43 +00:00 |
baihuawei
|
7cdbd3999e
|
fix gpu KLDivLoss st
|
2022-06-10 15:10:21 +08:00 |
zhujingxuan
|
2bf0a95a25
|
update InstanceNorm GPU kernel
|
2022-06-10 14:45:10 +08:00 |
i-robot
|
187b902cf3
|
!35737 takedown scatter_nd_update and sparse_segment_mean testcases
Merge pull request !35737 from yanghaoran/takedown_testcases
|
2022-06-10 06:44:02 +00:00 |
yanghaoran
|
42fee70ce0
|
takedown scatter_nd_update and sparse_segment_mean testcases
|
2022-06-10 14:32:26 +08:00 |
yanglf1121
|
dd7f6f91c0
|
support tensor to sparse tensor
|
2022-06-10 12:41:12 +08:00 |
i-robot
|
29a3c7f3ed
|
!34888 add Col2Im GPU impl
Merge pull request !34888 from baihuawei/col2im
|
2022-06-10 03:50:40 +00:00 |
zhangxuetong
|
86dc05b979
|
support functional tensor vmap for celu
|
2022-06-10 09:12:39 +08:00 |
baihuawei
|
404816a5f4
|
add gpu Col2Im impl and tensor & function op
|
2022-06-09 20:10:00 +08:00 |
hezhenhao1
|
17653575c5
|
Add random seed for SparseSegmentMean test case.
|
2022-06-09 17:43:37 +08:00 |
i-robot
|
598caa37ca
|
!35568 Extend the MatrixBandPart kernel to support batch.
Merge pull request !35568 from liqiliang/vmap_and_dynamic_and_tests
|
2022-06-09 06:14:48 +00:00 |
i-robot
|
490e502d01
|
!35557 add mish lrn lrn_grad vmap_rule and some test case
Merge pull request !35557 from zhuzhongrui/pub_master2
|
2022-06-09 03:29:02 +00:00 |
i-robot
|
4f07986eae
|
!35440 Add testcases for ScatterUpdate and unify the testing codes of ScatterMin/Max/Update
Merge pull request !35440 from zhengzuohe/test_scatterupdate
|
2022-06-09 03:20:59 +00:00 |
i-robot
|
1e6af66b5a
|
!35354 soft shrink
Merge pull request !35354 from ling/sr
|
2022-06-09 02:02:16 +00:00 |
i-robot
|
008953c4c0
|
!35466 tan support complex
Merge pull request !35466 from ling/bug
|
2022-06-09 01:39:27 +00:00 |
liqiliang
|
e0369d0d54
|
Extend the MatrixBandPart kernel to support batch.
|
2022-06-08 20:45:19 +08:00 |
i-robot
|
8c823836a6
|
!35383 Add cpu kernel for deformable_offsets
Merge pull request !35383 from YuJianfeng/deformable_conv
|
2022-06-08 11:13:30 +00:00 |
i-robot
|
aacf006016
|
!34766 add gpu deformable offset grad
Merge pull request !34766 from kisnwang/add-gpu-deformable-offset-grad
|
2022-06-08 09:35:03 +00:00 |
z00512249
|
c85cb977db
|
add mish lrn lrn_grad vmap_rule and some test case
|
2022-06-08 16:43:02 +08:00 |
i-robot
|
ddeaad24fd
|
!35386 add gpu dataformatdimmap
Merge pull request !35386 from KXiong/branch_gpu_dataformatdimmap
|
2022-06-08 08:35:11 +00:00 |
i-robot
|
dc6bee390e
|
!33385 UpsampleNearest3D GPU operator
Merge pull request !33385 from tom_chen/upsample
|
2022-06-08 07:52:04 +00:00 |
ling
|
60e7d1f947
|
tan support complex
|
2022-06-08 14:14:01 +08:00 |
yujianfeng
|
2d283c96df
|
Add the implement of cpu kernel for deformable_offsets
|
2022-06-08 11:53:52 +08:00 |
zhengzuohe
|
9cda5eac64
|
add testcases for ScatterUpdate and unify the testing codes of ScatterMin/Max/Update
|
2022-06-08 11:50:30 +08:00 |
ling
|
313274ae66
|
soft shrink
unsorted segment prod num_segments type check
|
2022-06-08 11:22:34 +08:00 |
xiongkun
|
7249a25ad7
|
add gpu dataformatdimmap
tick
add dataformatdimmap
add dataformatdimmap
add dataformatdimmap
add dataformatdimmap
add dataformatdimmap
|
2022-06-08 09:43:07 +08:00 |
kswang
|
410402ba6b
|
add gpu deformable offset grad
|
2022-06-08 09:09:05 +08:00 |
i-robot
|
4f97ed5fb0
|
!35550 Fix the mish_grad test case.
Merge pull request !35550 from liqiliang/mishgrad-cpu
|
2022-06-08 00:00:48 +00:00 |
tom__chen
|
b4c0635d53
|
upsample_nearest_3d GPU operator
|
2022-06-07 10:43:53 -04:00 |
i-robot
|
e84b1b3197
|
!35351 add determinant data_type && add vmap rule
Merge pull request !35351 from zhuzhongrui/pub_master
|
2022-06-07 13:17:24 +00:00 |
i-robot
|
6fe64a18c9
|
!34770 Add gpu DeformableOffsets
Merge pull request !34770 from tanghuikang/gpu_deformable_offset
|
2022-06-07 10:55:50 +00:00 |
i-robot
|
1617f280d3
|
!33752 Add the GPU version of GridSampler2D/GridSampler2DGrad and GridSampler3D/GridSampler3DGad operators
Merge pull request !33752 from dessyang/grid_sampler_gpu_ops_dev
|
2022-06-07 10:54:50 +00:00 |
i-robot
|
911b7b9950
|
!33331 gpu priority replay buffer implement
Merge pull request !33331 from chenweifeng/priority-replay-buffer-gpu-kernel
|
2022-06-07 10:39:57 +00:00 |
i-robot
|
8f51b50547
|
!35393 Add tensor and functional interface for ScatterNdMul and ScatterNdMax operator.
Merge pull request !35393 from hezhenhao1/add_scatter_max
|
2022-06-07 09:49:34 +00:00 |
liqiliang
|
6409772e72
|
Fix the mish_grad test case.
|
2022-06-07 17:18:17 +08:00 |
wilfChen
|
bea37279ff
|
gpu priority replay buffer
|
2022-06-07 16:08:08 +08:00 |
z00512249
|
cb1f8886e0
|
add determinant data_type && add vmap rule
|
2022-06-07 14:21:45 +08:00 |
i-robot
|
568278684d
|
!34540 UpsampleTrilinear3D GPU Kernel
Merge pull request !34540 from danishfarid/UpsampleTrilinear3D
|
2022-06-07 06:07:34 +00:00 |
i-robot
|
06824b4a2e
|
!35316 gpu_testcase_level_adjust
Merge pull request !35316 from LLLRT/master
|
2022-06-07 02:09:14 +00:00 |
i-robot
|
7b5287ffc7
|
!35475 fix scatter max gpu kernel and code check
Merge pull request !35475 from polyhedral/max
|
2022-06-07 02:00:23 +00:00 |
i-robot
|
74b9ef095e
|
!35322 Adapt MatrixDiagPartV3 GPU to new API
Merge pull request !35322 from zhujingxuan/MatrixDiagPart
|
2022-06-07 01:49:49 +00:00 |
dessyang
|
801763172a
|
Add GPU version of GridSampler2D/GridSampler3D and GridSampler2DGrad/GridSampler3DGrad operators
adapt to new NativeGpuKernelMod interface
|
2022-06-06 09:54:16 -04:00 |
hezhenhao1
|
bf6340504d
|
Add functional doc and interface for ScatterNdMul and ScatterNdMax operator.
|
2022-06-06 20:39:02 +08:00 |
tanghuikang
|
3e0b3a2ce0
|
Add gpu DeformableOffsets
|
2022-06-06 19:17:04 +08:00 |
polyhedral
|
bd7278de51
|
fix scatter max gpu kernel and code check
|
2022-06-06 18:58:38 +08:00 |
i-robot
|
621b48760b
|
!35347 flatten support functional and tensor interface, dynamic shape and vmap
Merge pull request !35347 from polyhedral/flatten
|
2022-06-06 09:50:51 +00:00 |
zhujingxuan
|
40899613ca
|
adapt MatrixDiagPartV3 GPU to new API
|
2022-06-06 17:33:18 +08:00 |