Megvii Engine Team
|
bb872965e3
|
feat(mge/module): add local conv2d module
GitOrigin-RevId: bb5ab94b19
|
5 years ago |
Megvii Engine Team
|
980ebf2c72
|
feat(mge/module): add fused conv_bn qat approximate version
GitOrigin-RevId: 1b7284a595
|
5 years ago |
Megvii Engine Team
|
b2ed2630f7
|
Revert "fix(mge/functional): rm useless arguments"
This reverts commit 8e4f25bfd8 .
GitOrigin-RevId: df696ab8a2
|
5 years ago |
Megvii Engine Team
|
d23fec06cb
|
fix(mge/functional): rm useless arguments
group and compute_mode are not used by local_conv2d
GitOrigin-RevId: 8e4f25bfd8
|
5 years ago |
Megvii Engine Team
|
814a427810
|
feat(mge/module): add `with_parent` argument in `_flatten`
GitOrigin-RevId: 1c88559ece
|
5 years ago |
Megvii Engine Team
|
dd962a4700
|
fix(mge/module): add negative slope attribute for LeakyReLU module
GitOrigin-RevId: 67e1967b56
|
5 years ago |
Megvii Engine Team
|
712b87c8c1
|
feat(mgb/core): add comp node for cambricon
add testcase for cambricon comp node
GitOrigin-RevId: 7794faa47f
|
5 years ago |
Megvii Engine Team
|
e8494f729b
|
fix(unittest): remove the torch dependency in test_conv
GitOrigin-RevId: 36dcab2b95
|
5 years ago |
Megvii Engine Team
|
7ed98dd50a
|
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
|
5 years ago |
Megvii Engine Team
|
dd41e2e47a
|
fix(mge/test):fix integration/manual tests
GitOrigin-RevId: 5933392407
|
5 years ago |
Megvii Engine Team
|
65432d3b93
|
fix(mge/module): fix torch subgraph under jit.trace with symbolic=False
GitOrigin-RevId: a208ba79d9
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |