Megvii Engine Team
|
75c413e3bb
|
feat(imperative): region restrictd conv support bias in python
GitOrigin-RevId: 9a2c1ee27a
|
2 years ago |
Megvii Engine Team
|
9e020d23b4
|
feat(imperative,mgb): add RegionRestrictedConvolution megbrain and imperative opr
GitOrigin-RevId: c0106ade08
|
2 years ago |
Megvii Engine Team
|
b50947d2e2
|
docs(mge/module): add Module.state_dict docstring
GitOrigin-RevId: 424ec791d7
|
2 years ago |
Megvii Engine Team
|
6e54a3bf31
|
docs(mge): update MaxPool2d & functional.nn.max_pool2d docstring
GitOrigin-RevId: b46f227b59
|
2 years ago |
Megvii Engine Team
|
98460f583f
|
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
|
2 years ago |
Megvii Engine Team
|
b2959589fc
|
docs(mge): update AvgPool2d example
GitOrigin-RevId: 2d5e515055
|
2 years ago |
Megvii Engine Team
|
ada54ba5a0
|
docs(mge): update AvgPool2d & functional.nn.avg_pool2d example
GitOrigin-RevId: e2859c7f99
|
2 years ago |
Megvii Engine Team
|
d8ea01680c
|
docs(mge): update AvgPool2d & functional.nn.max_pool2d docstring
GitOrigin-RevId: d69c5b024f
|
2 years ago |
Megvii Engine Team
|
99309fa3d6
|
feat(mge/functional): add param output_padding for deconv ops
GitOrigin-RevId: 8a69608953
|
2 years ago |
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 |