Commit Graph

101 Commits

Author SHA1 Message Date
i-robot 0a7e9ac87b
!35395 Add functional and tensor interface for `GatherD`
Merge pull request !35395 from TronZhang/gatherd_interface
2022-06-10 09:22:38 +00:00
i-robot 402431e754
!35667 [kernel]add celu vmap
Merge pull request !35667 from 张学同/celu
2022-06-10 08:21:43 +00: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
tronzhang 9a2fc5785d add functional and tensor interface for gatherd 2022-06-09 18:00:08 +08:00
looop5 4c5de144ce add functional, tensor interface and vmap for ops.Pad 2022-06-09 14:26:36 +08:00
i-robot eafd35217a
!33569 add hswish operator
Merge pull request !33569 from jjfeing/add_hswish_operator
2022-06-09 06:19:38 +00:00
i-robot d79a264e81
!35417 add masked_fill api
Merge pull request !35417 from 范吉斌/maskedfill_docs
2022-06-09 03:27:34 +00:00
jjfeing 5f8af95bb6 add hardswish & hardswishgrad 2022-06-09 10:51:38 +08:00
fanjibin dc6d60ffad add masked_fill api 2022-06-08 19:55:38 +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 0f27520ddb
!35560 Move ClipByNorm to inner interface
Merge pull request !35560 from JoyLvliang/move_clipbynorm_to_inner_interface
2022-06-08 09:27:05 +00:00
yujianfeng 2d283c96df Add the implement of cpu kernel for deformable_offsets 2022-06-08 11:53:52 +08:00
i-robot a27f5fcd22
!34594 add adaptive_avgpol2d nn/function/tensor api
Merge pull request !34594 from 范吉斌/adapool2d
2022-06-08 02:37:01 +00:00
7347157+joylvliang@user.noreply.gitee.com e8bc5e59e7 move_clipbynorm_to_inner_interface 2022-06-07 21:01:04 +08:00
i-robot e85d0cc246
!35227 gumbel softmax
Merge pull request !35227 from TuDouNi/gumbelsoftmax2
2022-06-07 10:59:53 +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
fan-jibin 624706d54b add adaptive_avgpol2d nn/function/tensor api 2022-06-07 14:48:55 +08:00
i-robot 375196215f
!35374 doc for renorm op
Merge pull request !35374 from liubuyu/doc
2022-06-07 02:50:40 +00:00
hezhenhao1 bf6340504d Add functional doc and interface for ScatterNdMul and ScatterNdMax operator. 2022-06-06 20:39:02 +08:00
i-robot 8968b238de
!35373 Update the doc of ScatterUpdate's functional interface
Merge pull request !35373 from zhengzuohe/doc_for_scatterupdate
2022-06-06 11:13:18 +00: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
lby 00c378dc92 doc for renorm 2022-06-06 17:23:07 +08:00
ttudu 4885dc69cd gumbel softmax 2022-06-06 17:09:40 +08:00
zhengzuohe 5b2701bbfe udpate docs for functional api of ScatterUpdate 2022-06-06 11:44:35 +08:00
fanjibin 938a17d314 add intopk vmap, functional/tensor api 2022-06-05 13:18:51 +08:00
polyhedral 92b1459d57 flatten support functional and tensor interface, dynamic shape and vmap 2022-06-04 11:49:55 +08:00
liqiliang ba67f2362b Add api docs. 2022-06-02 17:11:40 +08:00
i-robot f3251db54d
!35379 move logsumexp from element-wise to reduction
Merge pull request !35379 from 吕昱峰(Nate.River)/code_docs_logsumexp
2022-06-02 08:08:16 +00:00
i-robot f5ad517c23
!34925 add docs for scatter_div operator
Merge pull request !34925 from hujiahui8/code_docs_scatter_div
2022-06-02 08:01:21 +00:00
lvyufeng ccd67fa5d6 move logsumexp from element-wise to reduction 2022-06-02 15:57:53 +08:00
i-robot ad8dab99a3
!35361 add ops.logsumexp and fix nan problem
Merge pull request !35361 from 吕昱峰(Nate.River)/logsumexp
2022-06-02 07:50:12 +00:00
i-robot a03d31e8ed
!35302 add nll_loss and cross_entropy for ops.functional and nn
Merge pull request !35302 from 吕昱峰(Nate.River)/nllloss
2022-06-02 07:20:14 +00:00
i-robot 05afdee563
!34757 add inplace_update operator and vmaprule
Merge pull request !34757 from wangjun/inplace_ops_0523
2022-06-02 05:52:09 +00:00
lvyufeng cf067edc98 add ops.logsumexp and fix nan problem 2022-06-02 11:45:58 +08:00
lvyufeng ee30ffc3bf add nll_loss and cross_entropy for ops.functional and nn 2022-06-02 11:13:34 +08:00
hujiahui8 33a9fd87d3 add docs for scatter_div operator 2022-06-01 17:49:14 +08:00
wangjun 1f1857c51c add inplace update 2022-06-01 13:50:17 +08:00
z30020733 942d49e693 Add broadcast_to op api 2022-05-31 10:37:43 +08:00
i-robot 2734f2cca9
!34582 Add tensor&functional interface for TensorScatterMin ops.
Merge pull request !34582 from liangzelang/tensor_scatter_min
2022-05-30 07:42:46 +00:00
i-robot 256303c6ef
!35004 AdaptiveMaxPool2D op.
Merge pull request !35004 from gaoyong10/dynamic_shape_04
2022-05-30 07:36:24 +00:00
jiangshuqiang 0449f91f47 add bernoulli ops include func, Tensor interface, GPU kernel and docs. 2022-05-29 13:38:06 +08:00
gaoyong10 cd044f7446 AdaptiveMaxPool2d op. 2022-05-28 15:34:27 +08:00
i-robot 96bfdbbd68
!34998 fix doc string
Merge pull request !34998 from hezhenhao1/code_docs
2022-05-28 07:06:59 +00:00
i-robot a19c2f806c
!34995 update vmap doc
Merge pull request !34995 from Erpim/code_docs_vmap
2022-05-28 03:39:54 +00:00
i-robot 2f71ed63bb
!35046 Support scalar tensor as input of ClipByNorm
Merge pull request !35046 from JoyLvliang/fix_scalar_tensor_calculate_wrong_in_ClipByNorm
2022-05-28 03:18:59 +00:00
hezhenhao1 59a6dcfa26 fix doc string 2022-05-28 10:41:55 +08:00
7347157+joylvliang@user.noreply.gitee.com 6ebe248549 fix_scalar_tensor_calculate_wrong_in_ClipByNorm 2022-05-27 15:36:48 +08:00
Erpim fd40bff8f8 update vmap doc 2022-05-27 14:30:28 +08:00