kxz@thumt102-1
|
8f48da7ffe
|
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
|
3 years ago |
Megvii Engine Team
|
0c3699ff55
|
test(imperative/python): fix testcase for magicmind runtime module
GitOrigin-RevId: baf2f72f01
|
3 years ago |
Megvii Engine Team
|
f398c8e619
|
feat(imperative): add magicmind runtime opr
GitOrigin-RevId: 02ddb886d9
|
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
|
ea3eb66e7f
|
docs(mge): modify some docstring about conv and tools
GitOrigin-RevId: 4833e78d71
|
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
|
f2f3356572
|
fix(mge): fix some minor problems
GitOrigin-RevId: 43abda2ab9
|
3 years ago |
Megvii Engine Team
|
ac86d64474
|
fix(mge/module): fix prelu error when use_symbolic_shape is true
GitOrigin-RevId: 25b9c4d41d
|
3 years ago |
Megvii Engine Team
|
9d05738d54
|
Merge pull request #225 from chenjiahui0131:pixel_shuffle_module
GitOrigin-RevId: d94533655a
|
3 years ago |
Megvii Engine Team
|
8d507cc39d
|
docs(mge/module): add note about definition of momentum in BatchNorm
GitOrigin-RevId: 30a9aec20f
|
3 years ago |
chenjiahui
|
c8a9a3fd0a
|
feat(imperative): add PixelShuffle to module.__init__.py
|
3 years ago |
Megvii Engine Team
|
f7d05db7d6
|
fix(mge/module): fix named_tensors
GitOrigin-RevId: bb5aa1f41d
|
3 years ago |
Megvii Engine Team
|
2108696588
|
feat(mge/imperative): add local response normalization
GitOrigin-RevId: 939a4d26dd
|
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
|
27f2ecc436
|
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
|
3 years ago |
Megvii Engine Team
|
1019865071
|
feat(module): add tensors and named_tensors
GitOrigin-RevId: cb56d65d38
|
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
|
298c44433f
|
fix(mge/module): fix mistake warning message of module auto naming
GitOrigin-RevId: 4b85c13e90
|
3 years ago |
Megvii Engine Team
|
3eb0505f9b
|
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
|
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
|
186bacfb71
|
fix(mge): recover bn freeze fastpath execution
GitOrigin-RevId: 13e58c9fba
|
3 years ago |
Megvii Engine Team
|
e6caa9ff89
|
feat(opr): add bn backward for inference mode
GitOrigin-RevId: bb643cb62f
|
4 years ago |
Megvii Engine Team
|
b2944559a8
|
fix(imperative/module): remove ``__getattribute__`` method in module
GitOrigin-RevId: 5ac525f010
|
4 years ago |
Megvii Engine Team
|
f40df60242
|
docs(mge): refactor docs to remove warnings
GitOrigin-RevId: efefc2a4a2
|
3 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
Megvii Engine Team
|
0c97b2a3ec
|
fix(module): remove assert during forward
GitOrigin-RevId: bb04d861a3
|
3 years ago |
Megvii Engine Team
|
427113088b
|
fix(module/normalization): fix bug of LayerNorm and support input of any shape
GitOrigin-RevId: fd643addb5
|
4 years ago |
Megvii Engine Team
|
d6db4fea93
|
fix(mge/module): set no_cache=true when loading state dict
GitOrigin-RevId: 83281a3d47
|
4 years ago |
Megvii Engine Team
|
e275dfeca1
|
feat(imperative/python): support pooling mode "average" for avg pool2d module
GitOrigin-RevId: 9fe442129f
|
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
|
8be70d170c
|
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
|
4 years ago |
Megvii Engine Team
|
f31752d5cc
|
feat(mge/module): add __repr__ method for qat and quantized module
GitOrigin-RevId: 0d78cbab9a
|
4 years ago |
Megvii Engine Team
|
05e4c8266d
|
fix(mge/module): fix a mistaken in BN docs
GitOrigin-RevId: 086a70649e
|
4 years ago |
Megvii Engine Team
|
e18afa0b05
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
93bfda5191
|
fix(mge/module): fix module attribute update mistake
GitOrigin-RevId: 842a8fcb6e
|
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
|
af62cf6f05
|
fix(mge/utils): fix auto naming bug when expanding structure
GitOrigin-RevId: 5735bf5151
|
4 years ago |
Megvii Engine Team
|
aed681d325
|
feat(imperative/utils): optimize the naming rules
GitOrigin-RevId: 329bac640a
|
4 years ago |
Megvii Engine Team
|
dc96f6aa49
|
fix(mge/quantization): fix quantized concat forward problem
GitOrigin-RevId: dc21b340d1
|
4 years ago |
Megvii Engine Team
|
48a7d333c3
|
fix(mge/module): fix auto-naming error when there are containers in the module
GitOrigin-RevId: 6b8f043321
|
4 years ago |
Megvii Engine Team
|
c8697a7005
|
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
|
4 years ago |
Megvii Engine Team
|
5a38ad3920
|
feat(mge/utils): add get/set_expand_structure to deal with complex key
GitOrigin-RevId: 4d1b952068
|
4 years ago |