Megvii Engine Team
|
8a222d2c9c
|
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
|
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
|
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
|
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
|
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
|
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
|
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
|
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
|
3f3a256e0f
|
fix(mge/functional): fix conv* dtype promotion
GitOrigin-RevId: 3f03790cfc
|
3 years ago |
Megvii Engine Team
|
f76a2cc2c6
|
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
|
3 years ago |
Megvii Engine Team
|
070c811732
|
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
|
3 years ago |
Megvii Engine Team
|
5cef74a77e
|
feat(mge/amp): add GradScaler support
GitOrigin-RevId: 0ab4910360
|
4 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
75eb04c559
|
feat(mge/experimental): add WeightScaler support
GitOrigin-RevId: a1d5b68498
|
4 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
Megvii Engine Team
|
5e54821845
|
docs(syncbn): complete syncbn document
GitOrigin-RevId: 9b8d546f0b
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
7f9d441cbb
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
e18afa0b05
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
65af9cfcdc
|
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
|
4 years ago |
Megvii Engine Team
|
d0aa9b41ee
|
refactor(mge/functional): move nvof to vision, compatible with old usage
GitOrigin-RevId: cd9d9b4f5a
|
4 years ago |
Megvii Engine Team
|
e9982d613f
|
chore(mge/functional): add compatible code for functional api
GitOrigin-RevId: 3b2f829cc5
|
4 years ago |
Megvii Engine Team
|
8cfed4a1cb
|
refactor(functional/nn): remove nn.svd
GitOrigin-RevId: 4f58e10a1f
|
4 years ago |
Megvii Engine Team
|
d970b85deb
|
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
|
4 years ago |
Megvii Engine Team
|
dcfb6a537e
|
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
|
4 years ago |
Megvii Engine Team
|
fe9c6e26a8
|
feat(imperative/opr): deprecate resize op and make it as a special case of interpolate
GitOrigin-RevId: a5668c5779
|
4 years ago |
Megvii Engine Team
|
1d7dd00144
|
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
|
4 years ago |
Megvii Engine Team
|
4f3875eb4f
|
docs(mge): add missing docstring and fix sphinx build warnings
GitOrigin-RevId: 4ce73cfd80
|
4 years ago |
Megvii Engine Team
|
98d6ba7e2f
|
refactor(mge/functional): move matinv from nn to math
GitOrigin-RevId: 75c48ce327
|
4 years ago |
Megvii Engine Team
|
5b697c714c
|
chore(imperative): remove unused reference to megbrain_graph
GitOrigin-RevId: 2c31f13bb8
|
4 years ago |
Megvii Engine Team
|
bb8f292816
|
refactor(mge/func): change `conv_execution_strategy` to `execution_strategy`
GitOrigin-RevId: 7aef9935ee
|
4 years ago |
Megvii Engine Team
|
05692d058f
|
refactor(functional): move matmul dot svd to math pkg
GitOrigin-RevId: 15eb08bacb
|
4 years ago |
Megvii Engine Team
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
4 years ago |
Megvii Engine Team
|
e82fa4ec23
|
fix(gopt): using new_inp for build_chain in DelayBroadcast pass
GitOrigin-RevId: efc6377197
|
4 years ago |
Megvii Engine Team
|
e3a3e0cd3c
|
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
4aaae9950b
|
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
|
4 years ago |
Megvii Engine Team
|
df976782fa
|
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
|
4 years ago |
Megvii Engine Team
|
ca1d74ad3f
|
fix(mge/utils): fix types import error
GitOrigin-RevId: d2d4bd2725
|
4 years ago |
Megvii Engine Team
|
e9a7be46b1
|
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
|
4 years ago |