Megvii Engine Team
f398c8e619
feat(imperative): add magicmind runtime opr
GitOrigin-RevId: 02ddb886d9
3 years ago
Megvii Engine Team
3d6fe33e2d
Revert "fix(imperative): check async error when getting value"
This reverts commit 3945a9bfa2
GitOrigin-RevId: ec6d3574b2
3 years ago
Megvii Engine Team
57e197475b
fix(imperative): check async error when getting value
GitOrigin-RevId: 3945a9bfa2
3 years ago
Megvii Engine Team
7ac693a882
docs(imperative): fix multiple target warning when cross-reference
GitOrigin-RevId: c5b4884c8a
3 years ago
Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
3 years ago
Megvii Engine Team
ea3eb66e7f
docs(mge): modify some docstring about conv and tools
GitOrigin-RevId: 4833e78d71
3 years ago
Megvii Engine Team
acd835d0df
docs(mge): add some docs for cvt_color and GraphInference
GitOrigin-RevId: d78628bfcf
3 years ago
Megvii Engine Team
5c94db16f2
feat(mge/functional): add groups support for conv_transpose2d & 3d
GitOrigin-RevId: b75b792fb4
3 years ago
Megvii Engine Team
8a222d2c9c
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
3 years ago
Megvii Engine Team
27aa648ba7
fix(mge): control compute_mode by context
GitOrigin-RevId: f3968309f4
3 years ago
Megvii Engine Team
846cc13ca8
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
3 years ago
MegChai
63bb7c2946
docs(mge/functional): update functional.arange docstring
From GITHUB 386
COPYBARA_INTEGRATE_REVIEW=https://github.com/MegEngine/MegEngine/pull/386 from MegChai:docstring-arange 6702c6a5cd
Closes #386
GitOrigin-RevId: e27cdaefe8
3 years ago
Megvii Engine Team
936bb23738
fix(mge/tensor): fix is/is not SyntaxWarning occurs in python3.8
GitOrigin-RevId: 12b66da367
3 years ago
Megvii Engine Team
29f9935d92
fix(imperative/python): add layer_norm doc and rm useless param
GitOrigin-RevId: 1b15db621e
3 years ago
Megvii Engine Team
2108696588
feat(mge/imperative): add local response normalization
GitOrigin-RevId: 939a4d26dd
3 years ago
Megvii Engine Team
ca4c93dee7
fix(mge): fix layer norm amp bug
GitOrigin-RevId: dba691fcbf
3 years ago
Megvii Engine Team
cebda6ff8d
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
3 years ago
Megvii Engine Team
f5cb21ed3a
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
3 years ago
Megvii Engine Team
c0ccd0ea7e
feat(mge/bn): add NHWC support for bn
GitOrigin-RevId: 0a5bb6f72d
3 years ago
chenjiahui
6b863cc53a
feat(imperative): add pixel_shuffle opr
Closes #211
GitOrigin-RevId: 1846714791
3 years ago
Megvii Engine Team
97b1b7774d
feat(mgb): add megbrain layer norm opr with subgraph
GitOrigin-RevId: 9b7fa821f8
3 years ago
Megvii Engine Team
43fb2c4afb
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
3 years ago
Megvii Engine Team
4688bbabad
refactor(functional): hide functional.vision and replace with functional.nn
GitOrigin-RevId: b6888c7f65
3 years ago
Megvii Engine Team
8796586b89
refactor(functional): import all from metric in nn
GitOrigin-RevId: 41ab78d78d
3 years ago
Megvii Engine Team
27f2ecc436
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
3 years ago
Megvii Engine Team
bd817f3af2
perf(syncbn): fallback to bn when sync is not required
GitOrigin-RevId: e89fdd87b0
3 years ago
Megvii Engine Team
ffcb4dacd9
style(mge): change the pad opr doc and format
GitOrigin-RevId: 7b20e96519
3 years ago
Megvii Engine Team
43cb305e53
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
3 years ago
Megvii Engine Team
7a9f2ed95e
perf(functional/split): add python binding for Split opr
GitOrigin-RevId: 3359f5da26
3 years ago
Megvii Engine Team
ab309eb5fc
feat(mgb/opr): let Split support empty IO
GitOrigin-RevId: aad6dc06bf
3 years ago
Megvii Engine Team
8485eff12c
docs(mge/functional): add docs for megengine.functional.sub
GitOrigin-RevId: 40208739bd
3 years ago
Megvii Engine Team
e6a8b0256e
fix(mge): ignore errors caused by earlier async errors
GitOrigin-RevId: ce2028d38a
3 years ago
Megvii Engine Team
ee790cb62c
fix(imperative/python): fix warp_perspective doc and arange dtype
GitOrigin-RevId: dc4e021ec0
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
76ce81e828
fix(mge): fix F.nn.dropout train and inference bugs
GitOrigin-RevId: 9d9f246d7b
3 years ago
Megvii Engine Team
29d24dbb80
fix(mge/function): fix interpolate unsupport fp16 error
GitOrigin-RevId: 7fc6271986
3 years ago
Megvii Engine Team
67575d582c
feat(mge/opr): add interpolate bilinear mode
GitOrigin-RevId: f7023a3fd3
3 years ago
Megvii Engine Team
0558b2123d
feat(mge/opr): add interpolate nearest mode
GitOrigin-RevId: d384b87f50
3 years ago
Megvii Engine Team
d063d5774f
perf(functional): use fma to reduce elemwise but disable subgraph compilation
GitOrigin-RevId: c75a6e1a09
3 years ago
Megvii Engine Team
2a063f8e87
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
3 years ago
Megvii Engine Team
3206af9db2
perf(functional/matmul): reimplement matmul with subgraph
GitOrigin-RevId: 456b2a51d3
3 years ago
Megvii Engine Team
8c47c1f149
perf(syncbn): reimplement with subgraph
GitOrigin-RevId: 13e7e3d3c0
3 years ago
Megvii Engine Team
77309609fa
perf(functional/dropout): add fastpath for dropout
GitOrigin-RevId: 3bf8546908
3 years ago
Megvii Engine Team
3eb0505f9b
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
3 years ago
Megvii Engine Team
dd1fecdf29
feat(mge/opr): add cumsum
GitOrigin-RevId: 740f00a8e5
3 years ago
Megvii Engine Team
7138e4fd02
feat(docs): add docs for megengine.functional.full
GitOrigin-RevId: 8428a0c9a9
3 years ago
Megvii Engine Team
e3fc783642
fix(mgb/opr): fix nvof shape error
fix(mgb/opr): fix format error
fix(mgb/opr): update nvof opr to 1.3 fix (compared to 2.0)
fix(mgb/opr): add try catch to solve test error
GitOrigin-RevId: b688745b36
3 years ago
Megvii Engine Team
3f3a256e0f
fix(mge/functional): fix conv* dtype promotion
GitOrigin-RevId: 3f03790cfc
3 years ago
Megvii Engine Team
536506c3f4
feat(functional): let interpolate support more modes
GitOrigin-RevId: 9693a1ac63
3 years ago
Megvii Engine Team
f76a2cc2c6
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
3 years ago