Commit Graph

84 Commits

Author SHA1 Message Date
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
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
liangzelang 8ace2766c4 Add tensor&functional interface for TensorScatterMin. 2022-05-27 09:37:31 +08:00
hemaohua cd9c40f807 add pdist_cpu_kernel 2022-05-27 09:01:07 +08:00
i-robot b62d4d84d1
!34970 add tensor_scatter_mul doc
Merge pull request !34970 from zhuzhongrui/code_docs_scipy
2022-05-26 09:59:03 +00:00
i-robot 67c7712c9d
!34987 add lp_norm doc
Merge pull request !34987 from zhuzhongrui/code_docs_branch1
2022-05-26 09:58:18 +00:00
z00512249 59643981ba add lp_norm 2022-05-26 16:49:00 +08:00
z00512249 a07f4b1bf9 add tensor_scatter_mul doc 2022-05-26 15:56:44 +08:00
yangruoqi713 6e32fe1fb2 [OP] modify docs for hardshrink, meshgrid 2022-05-26 14:13:10 +08:00
yelihua 79316f6caf add GPU impl for Fills and testcase 2022-05-25 18:29:10 +08:00
i-robot 5dfd613839
!34447 Support GPU UniqueConsecutive, and add tensor, functional, primitive
Merge pull request !34447 from huangbingjian/unique_consecutive
2022-05-25 09:19:35 +00:00
huangbingjian 5a619651d4 Add unique_consecutive ops. 2022-05-24 18:22:07 +08:00
polyhedral fd411c3144 ScatterMax support vmap and add dynamic shape case 2022-05-24 10:27:49 +08:00
yangzhenzhang 3b187616e7 add vmap for masked select 2022-05-23 09:10:35 +08:00
i-robot 3aeb0d0578
!34625 add tensor/functional interface for diag op
Merge pull request !34625 from limingqi107/bug_fix4
2022-05-20 20:36:04 +00:00
z00512249 7ec5d95d9d fix scatter_nd_add && scatter_nd_sub doc api 2022-05-20 15:35:12 +08:00
limingqi107 817e60a308 add tensor/functional interface for diag op 2022-05-20 15:05:11 +08:00
i-robot e1790a2e09
!34641 add the docs of bessel_k1 and bessel_i1
Merge pull request !34641 from shenghong96/doc_bessel
2022-05-20 01:18:14 +00:00
shenghong96 b591fcf63c add the docs of bessel i1 i1e and bessel k1 k1e 2022-05-19 18:20:12 +08:00