Commit Graph

477 Commits

Author SHA1 Message Date
xulei 5b170a5ac4 update docs for KLDivLoss 2022-06-13 10:22:31 +08:00
i-robot 0f160775c5
!35618 Update ms_kernel custom op
Merge pull request !35618 from zichun_ye/custom_op_update
2022-06-13 02:00:54 +00:00
i-robot 1766015d77
!35619 code style
Merge pull request !35619 from liubuyu/doc
2022-06-13 01:34:36 +00:00
panshaowu fa32247b6f add cpu operation of StandardLaplace and its functional interface 2022-06-11 18:25:58 +08:00
chenzhuo 2529bd3c2c add cpu ops Argmin add support Ascend dynamic ArgminV2 2022-06-11 17:20:47 +08:00
i-robot d6bbef46ee
!35448 add deformable offsets grad op for Ascend
Merge pull request !35448 from guoqi/deformable_offsets_grad
2022-06-11 07:39:42 +00:00
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 91a7ae0de0
!35646 make clear that Ger doesn't support mixed dtype input
Merge pull request !35646 from 李林杰/0609fix
2022-06-11 02:47:55 +00:00
lilinjie 9a8c0f825c some small fixes 2022-06-10 19:58:52 +08:00
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
Zichun Ye 8de24b4d2f update ms_kernel related files
fix import problem

update code check

update test case

update test case

rename file
2022-06-10 15:41:38 +08:00
guoqi b68e7d5045 add deformable offset grad operator for Ascend 2022-06-10 15:14:17 +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
i-robot 260f2bda0f
!35490 add functional, tensor interface and vmap for ops.Pad
Merge pull request !35490 from looop5/op_pad_commit
2022-06-10 01:29:10 +00:00
zhangxuetong 86dc05b979 support functional tensor vmap for celu 2022-06-10 09:12:39 +08:00
lby 968531f2f1 code style 2022-06-10 08:34:11 +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
yujianfeng 02626022a2 Fix the formula in the docs of deformable_conv2d 2022-06-09 15:07:55 +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 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 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
i-robot f2acc43307
!35558 fix intopk docs
Merge pull request !35558 from 范吉斌/fix_intopk_docs
2022-06-09 01:53:37 +00:00
liqiliang e0369d0d54 Extend the MatrixBandPart kernel to support batch. 2022-06-08 20:45:19 +08:00
fanjibin dc6d60ffad add masked_fill api 2022-06-08 19:55:38 +08:00
fan-jibin 19f551149a fix intopk docs 2022-06-08 19:31:52 +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
i-robot 37f115a0ef
!35605 modify format
Merge pull request !35605 from 俞涵/code_docs_0606
2022-06-08 07:12:16 +00:00
huodagu 85a5e03e9a modify format 2022-06-08 14:40:53 +08:00
i-robot b76a64d1fa
!35554 update pdist docs
Merge pull request !35554 from hemaohua/master
2022-06-08 06:34:38 +00:00
i-robot 3db02540ad
!35529 Update Raises in ScatterUpdate doc
Merge pull request !35529 from zhengzuohe/doc_for_scatterupdate
2022-06-08 04:03:23 +00:00
yujianfeng 2d283c96df Add the implement of cpu kernel for deformable_offsets 2022-06-08 11:53:52 +08:00
lvyufeng f4f2f2fc6f fix desciption problems 2022-06-08 11:22:25 +08:00
i-robot 964e01ddb0
!35528 modify format
Merge pull request !35528 from 俞涵/code_docs_0606
2022-06-08 02:50:35 +00:00
i-robot be5fd85eae
!35585 Fix GradOperation document issues.
Merge pull request !35585 from 张清华/code_docs_opt
2022-06-08 02:38:51 +00: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
Zhang Qinghua 5683ce2d81 Fix GradOperation document issues. 2022-06-08 10:31:04 +08:00
i-robot 275ee3c637
!35459 Fix GradOperation descripition and support tuple/list for 'get_by_list'.
Merge pull request !35459 from 张清华/opt
2022-06-08 01:46:39 +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
hemaohua 684853f7e3 update pdist docs 2022-06-07 18:30:15 +08: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
huodagu 067750a0bb modify format 2022-06-07 15:05:12 +08:00
zhengzuohe 6e3b8735e6 update Raises in ScatterUpdate doc 2022-06-07 15:00:13 +08:00
fan-jibin 624706d54b add adaptive_avgpol2d nn/function/tensor api 2022-06-07 14:48:55 +08:00
Zhang Qinghua ce39353421 Fix GradOperation descripition and support tuple/list for 'get_by_list'. 2022-06-07 14:21:58 +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
i-robot 62653a9697
!35471 fix nllloss and crossentropy bug and description typo error
Merge pull request !35471 from 吕昱峰(Nate.River)/nllloss
2022-06-07 01:48: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
lvyufeng 9ec87173fe fix nllloss and crossentropy bug and description typo error 2022-06-06 17:26:10 +08: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
i-robot 43df0f985c
!35461 modify format
Merge pull request !35461 from 俞涵/code_docs_0606
2022-06-06 08:40:27 +00:00
huodagu 4fbf67236b modify format 2022-06-06 15:03:02 +08:00
i-robot 945aa1b1d3
!34597 add intopk vmap, functional/tensor api
Merge pull request !34597 from 范吉斌/intopk_vmap
2022-06-06 06:41:06 +00: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
huoxinyou e7c769eebd AdamApplyOne support dynamic shape &
ScatterNd ops's indices support int64 docs
2022-06-02 20:21:33 +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 b58f42b671
!35301 modify format
Merge pull request !35301 from 俞涵/code_docs_0601
2022-06-02 06:29:39 +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
i-robot 8ae7250a24
!35213 Add broadcast_to op api
Merge pull request !35213 from 朱家兴/master
2022-06-02 03:50:23 +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
i-robot ae0a4815f5
!35217 [DOC][OP]Fix space_to_batch_nd op doc
Merge pull request !35217 from zichun_ye/code_docs_space_to_batch_nd
2022-06-02 02:06:12 +00:00
hujiahui8 33a9fd87d3 add docs for scatter_div operator 2022-06-01 17:49:14 +08:00
i-robot 0a05f67681
!35296 Update api doc Ainh & Softsign
Merge pull request !35296 from panzhihui/code_docs_cn_api4
2022-06-01 09:08:10 +00:00
panzhihui 2dd576e8d6 update doc 2022-06-01 14:29:06 +08:00
wangjun 1f1857c51c add inplace update 2022-06-01 13:50:17 +08:00
huodagu b4cef0a3e6 modify format 2022-06-01 11:45:09 +08:00
yelihua 3e4b0fec5b fix the raises type format 2022-06-01 11:36:12 +08:00
i-robot 21bcf23bbd
!35215 Update svd api docs
Merge pull request !35215 from huanghui/code_docs-update-svd-api
2022-05-31 11:17:54 +00:00
i-robot d8547cc27e
!35232 modify format
Merge pull request !35232 from 俞涵/code_docs_0531
2022-05-31 09:15:11 +00:00
huodagu 797cf1ef1b modify format 2022-05-31 15:35:09 +08:00
i-robot e6e6d0633c
!35091 update pdist interface doc
Merge pull request !35091 from hemaohua/master
2022-05-31 06:39:40 +00:00
i-robot e682f0bb2b
!35169 Correct ClipByNorm api doc
Merge pull request !35169 from JoyLvliang/correct_clipbynorm_api_doc
2022-05-31 03:20:28 +00:00
Zichun Ye e67654b729 update doc string for the space_to_batch_nd op
fix typo

fix typo
2022-05-31 10:51:34 +08:00
z30020733 942d49e693 Add broadcast_to op api 2022-05-31 10:37:43 +08:00
huanghui 7a0f5f661b update svd api docs 2022-05-31 10:06:10 +08:00
hezhenhao1 3832cc2115 fix doc string 2022-05-30 16:34:23 +08:00
i-robot 63701b7070
!35166 add parameters
Merge pull request !35166 from 俞涵/code_docs_0530
2022-05-30 07:51:04 +00:00
7347157+joylvliang@user.noreply.gitee.com 8a176ca1ad correct_clipbynorm_api_doc 2022-05-30 15:49:52 +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
hemaohua 75fd564200 update pdist interface doc 2022-05-30 15:28:45 +08:00
huodagu 4d4102f1a3 add parameters for output_shapes 2022-05-30 15:19:16 +08:00
i-robot 51c1918011
!34849 add bernoulli ops include func, Tensor interface, GPU kernel and docs.
Merge pull request !34849 from jiangshuqiang/add_bernoulli_
2022-05-30 06:34:08 +00:00
i-robot fe95a4feca
!34966 optimizes the documentation of chinese API of Mod,MulNoNan,etc
Merge pull request !34966 from zhangfanghe/code_docs_master_14_526
2022-05-30 03:37:25 +00:00
zhangfanghe 85d4c2ef44 optimizes the documentation of chinese API of Mod,MulNoNan,etc. 2022-05-30 11:05:44 +08:00
i-robot 8ab56385bb
!35092 Add translated Chinese API document, e.g. ROIAlign, Softplus...
Merge pull request !35092 from panzhihui/code_docs_cn_api3
2022-05-30 02:40:32 +00:00
i-robot d41b1ab22d
!35105 tan docs
Merge pull request !35105 from ling/code_docs_sr
2022-05-30 01:43:18 +00:00
panzhihui 2d61667b09 Add translated CN API documents 2022-05-29 19:36:35 +08:00
jiangshuqiang 0449f91f47 add bernoulli ops include func, Tensor interface, GPU kernel and docs. 2022-05-29 13:38:06 +08:00
i-robot 44bf5724a9
!35112 modify format
Merge pull request !35112 from 俞涵/code_docs_0527
2022-05-28 08:28:38 +00:00
huodagu 797c7484a1 modify format 2022-05-28 16:07:00 +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
ling dc0d5185b8 tan docs 2022-05-28 14:46:34 +08: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
i-robot 35d2a0fdd0
!34911 optimizes the documentation of chinese API of FloorMod,Invert,etc
Merge pull request !34911 from zhangfanghe/code_docs_master_11_525
2022-05-28 02:29:07 +00:00
i-robot 012e69c911
!35060 fix lp_norm doc.
Merge pull request !35060 from zhuzhongrui/code_docs_branch1
2022-05-27 10:19:32 +00:00
i-robot 56462cc071
!34486 add pdist_cpu_kernel
Merge pull request !34486 from hemaohua/pdist_push
2022-05-27 09:54:58 +00:00
z00512249 33d6102040 fix lp_norm doc. 2022-05-27 16:14:24 +08:00
7347157+joylvliang@user.noreply.gitee.com 6ebe248549 fix_scalar_tensor_calculate_wrong_in_ClipByNorm 2022-05-27 15:36:48 +08:00
i-robot 4adf5d390c
!35047 modify format
Merge pull request !35047 from 俞涵/code_docs_0527
2022-05-27 07:08:12 +00:00
huodagu 11a8fef875 modify Probability_rst and format 2022-05-27 14:51:00 +08:00
i-robot 62f25c572d
!34851 Add translated Chinese API documents
Merge pull request !34851 from panzhihui/code_docs_cn_api
2022-05-27 06:46:48 +00:00
Erpim fd40bff8f8 update vmap doc 2022-05-27 14:30:28 +08:00
i-robot a61b57b921
!34944 Add Chinese API document translationsm, e.g. InTopK, KLDivLoss, Log1p
Merge pull request !34944 from panzhihui/code_docs_cn_api2
2022-05-27 06:02:29 +00:00
i-robot d5fa86fea0
!34717 Fix OneHot docs and vmap
Merge pull request !34717 from jiaoy1224/onehot
2022-05-27 03:38:26 +00:00
panzhihui 127cb7d969 Add chinese api doc translation, InTopk, KLDivLoss etc. 2022-05-27 11:37:35 +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
panzhihui 58fdd7e623 add cn api docs 2022-05-26 19:56:21 +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
i-robot 4380e9d974
!34993 add fills docs
Merge pull request !34993 from yelihua/code_docs_master
2022-05-26 09:47:02 +00:00
i-robot 2b2888593e
!34889 [OP] modify docs for hardshrink, meshgrid
Merge pull request !34889 from yangruoqi713/code_docs_op
2022-05-26 09:33:25 +00:00
zhangfanghe a0641550a8 optimizes the documentation of chinese API of FloorMod,Invert,etc. 2022-05-26 17:20:17 +08:00
huodagu a7f9e1a9d0 modify format 2022-05-26 17:07:25 +08:00
yelihua 9a9ab882e9 delete ops.Fills docs 2022-05-26 17:04:58 +08: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
i-robot bc5bb22343
!34733 optimizes the documentation of chinese API of ApplyAdagradDA,ApplyAdagradV2,etc
Merge pull request !34733 from zhangfanghe/code_docs_master_11_521
2022-05-25 09:12:18 +00:00
zhangfanghe cbb1367a59 optimizes the documentation of chinese API of ApplyAdagradDA,ApplyAdagradV2,etc. 2022-05-25 17:00:49 +08:00
i-robot 7c0de1a877
!34918 modify docs for masked select
Merge pull request !34918 from yangzhenzhang/code_docs_masked_select_en
2022-05-25 06:44:30 +00:00
i-robot 94cac732c8
!34723 Fix the description of mode parameter in the Conv2d API.
Merge pull request !34723 from panzhihui/code_docs_conv2d
2022-05-25 06:20:55 +00:00
yangzhenzhang 570a8304ec modify doc for masked select 2022-05-25 14:16:31 +08:00
i-robot 520427cc3a
!34726 ScatterMax support vmap and add dynamic shape case
Merge pull request !34726 from polyhedral/max
2022-05-25 02:14:45 +00:00
Yang Jiao 21c58913c7 fix onehot vmap 2022-05-25 09:36:28 +08:00
huangbingjian 5a619651d4 Add unique_consecutive ops. 2022-05-24 18:22:07 +08:00
huodagu c73490f063 modify format 2022-05-24 14:54:03 +08:00