Commit Graph

930 Commits

Author SHA1 Message Date
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