- 13 9月, 2021 10 次提交
-
-
由 Yiqun Liu 提交于
Implement FunctionTraits to support two kinds of elementwise functor and remove some old codes for broadcast. (#35487)
-
由 xiongkun 提交于
-
由 taixiurong 提交于
-
由 JZ-LIANG 提交于
* reshape support zero-input * add unitest * revise error message
-
由 李季 提交于
* upload global scatter and global gather operators related files
-
由 Zhang Zheng 提交于
-
由 Qi Li 提交于
-
由 Yanxing Shi 提交于
* fix github name * fix CI error * fix review and CI error * fix inf,nan error and modify unittest samples * add unittest samples * add unittest samples * fix unittest error * test=document_fix * test=document_fix * modify doc and add unittest samples * fix error newline in constant * modify doc after mentor review * modify __all__ and doc * modify doc
-
由 zhulei 提交于
* [ROCM] fix top_k_v2 with large shape * [ROCM] fix top_k_v2 with large shape
-
由 jakpiase 提交于
* implemented clip op bf16/fp32 * added skipping if not cpu or bf16 * CI rerun after bf16 package change * added parentheses to ensure formatting
-
- 12 9月, 2021 1 次提交
-
-
由 王明冬 提交于
-
- 11 9月, 2021 3 次提交
- 10 9月, 2021 13 次提交
-
-
由 Feng Xing 提交于
-
由 Feng Xing 提交于
-
由 zhouweiwei2014 提交于
-
由 zhiboniu 提交于
-
由 hlygit66666 提交于
* add test_cumprod_op * Revert "add test_cumprod_op" This reverts commit c96cf6dff5d09ae7d8cc72c1e8ae4369a153aa19. * recommit * add error message * test input(x) initialize * test use cpu * update test code * add test type * add test case * solve ci problem * add complex case test * add complex case test * fix review problem * fix conflict * fix some docs * change test case * change test case * fix review problems again * fix docs * fix inclusivescan bug
-
由 huangxu96 提交于
This PR supports gradient clip (ClipGradByGlobalNorm) when training with AMP(auto mixed precision).
-
由 ShenLiang 提交于
-
由 huangjun12 提交于
* add as_extra to dropout op and lrn op * refine details * fix dropout op maker
-
由 Jiawei Wang 提交于
* fix extra op for expand, expand_as, tile, unstack * Update expand_v2_op.cc
-
由 ceci3 提交于
* fix bn/in/squeeze/syncbn extra * update bn * update * update
-
由 Shang Zhizhou 提交于
* add opdef extra * add reduce mean * update style
-
由 Zeng Jinle 提交于
* fix scatter gather bug: * fix windows ci
-
由 pangyoki 提交于
* add asExtra for nce op * fix unittest error in macos * remove asExtra for is_test
-
- 09 9月, 2021 5 次提交
-
-
由 Wei Shengyu 提交于
* mark extra attr for unsqueeze2 * debug for inference
-
由 crystal 提交于
-
由 whs 提交于
-
由 zhangkaihuo 提交于
-
由 0x45f 提交于
* init matrix_rank op, add matrix_rank CPU code and test * add GPU kernel, remove svd_eigen.h * add CPU kernel when tol is tensor * add cpu and gpu code when tol is tensor * fix CI-ROCM error * add matrix_rank API describe, fix PR-CI-Py3 error * fix PR-CI-Windows error, add matrix_rank API test * delete useless comments * fix review * add my code in svd_helper.h * update doc commets * remove spaces
-
- 08 9月, 2021 8 次提交
-
-
由 Li Min 提交于
-
由 will-jl944 提交于
multiply supports bool
-
由 niuliling123 提交于
-
由 CtfGo 提交于
* mark WhileOp AsExtra attribute * revert kX and kOutputs
-
由 niuliling123 提交于
-
由 Leo Chen 提交于
* add clip_by_norm fp16 kernel * add ut
-
由 hong19860320 提交于
-
由 Guoxia Wang 提交于
-