Megvii Engine Team
|
fd41302cc1
|
feat(imperative/amp): add set_format
GitOrigin-RevId: 91de6f49de
|
3 years ago |
Megvii Engine Team
|
fc633ce4ff
|
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
|
3 years ago |
Megvii Engine Team
|
e393d1cf65
|
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
3f11b4213a
|
docs(functional): replace megengine function testcode with doctest format
GitOrigin-RevId: f91d5f5afd
|
3 years ago |
Megvii Engine Team
|
60163c5d12
|
docs(functional): replace elem testcode with doctest format
GitOrigin-RevId: a32a011724
|
3 years ago |
Megvii Engine Team
|
a4327c4d25
|
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
|
3 years ago |
Megvii Engine Team
|
90c8a58cca
|
docs(docstring): add pad docstring
GitOrigin-RevId: eaf6a87456
|
3 years ago |
Megvii Engine Team
|
ab6d12caff
|
feat(mge): add conv padding mode
GitOrigin-RevId: 147ced856e
|
3 years ago |
Megvii Engine Team
|
bebb2cf4c3
|
Merge pull request #428 from P2Oileen:fix-pad
GitOrigin-RevId: f33ea46ad6
|
3 years ago |
Megvii Engine Team
|
6157d9cfef
|
fix(traced_module): fix Module compatible issue and traced module getattr check
GitOrigin-RevId: 62eb3bfb10
|
3 years ago |
Megvii Engine Team
|
3159eecadd
|
fix(init): fix fan_in and fan_out for group conv2d
GitOrigin-RevId: a6f41063f0
|
3 years ago |
Megvii Engine Team
|
1d2510b6d7
|
fix(module): fix module dumped in old version without _short_name attr
GitOrigin-RevId: a1c815f613
|
3 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
3 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
3 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
P2Oileen
|
5d755898fd
|
fix(funtional): correct the typo: witdth to width
|
3 years ago |
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 |