Megvii Engine Team
|
888c7f1f7a
|
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 |
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
|
0d0968557f
|
fix(imperative/quantization): add conditional when passing parameters name
GitOrigin-RevId: f38042aeab
|
4 years ago |
Megvii Engine Team
|
2763846168
|
fix(mge/module): support list/dict/tuple in module __repr__
GitOrigin-RevId: b70193fd79
|
4 years ago |
Megvii Engine Team
|
ce88e6c4f7
|
feat(mge/quantization): use extra act_fakequant to decide whether to do bias fakequant
GitOrigin-RevId: bf54012155
|
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
|
a1ca50c923
|
feat(mge/quantization): add name for quantized module
GitOrigin-RevId: edefbec7b7
|
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
|
47db29aaa2
|
fix(mge/module): add kwargs param for all modules
GitOrigin-RevId: 7245e669a7
|
4 years ago |
Megvii Engine Team
|
6fb19b664b
|
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
|
4 years ago |
Megvii Engine Team
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
4 years ago |
Megvii Engine Team
|
dbb3dd681f
|
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
|
4 years ago |
Megvii Engine Team
|
27a9326c66
|
fix(mge/module): fix frozen batch norm
GitOrigin-RevId: 143d468a37
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
4 years ago |
Megvii Engine Team
|
ca1d74ad3f
|
fix(mge/utils): fix types import error
GitOrigin-RevId: d2d4bd2725
|
4 years ago |
Megvii Engine Team
|
abe3c165ba
|
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
|
4 years ago |
Megvii Engine Team
|
82847bc70a
|
docs(mge): add conv2d output shape computation formula
GitOrigin-RevId: 7ee476a80d
|
4 years ago |
Megvii Engine Team
|
cde9727a4f
|
fix(mge/quantization): fix `qat.conv_bn` parameter issue
GitOrigin-RevId: 4aa3836926
|
4 years ago |
Megvii Engine Team
|
94796060b6
|
test(mge/quantization): qat module, quantized op and module
GitOrigin-RevId: 17b28060cc
|
4 years ago |
Megvii Engine Team
|
ab9f44f15c
|
feat(mge/quantization): add support for easyquant
GitOrigin-RevId: 060d908349
|
4 years ago |
Megvii Engine Team
|
69e3e32240
|
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
|
4 years ago |
Megvii Engine Team
|
6c4841e807
|
fix(mge/quantization): `disable_fake_quant` does not work correctly
GitOrigin-RevId: 0c568d3335
|
4 years ago |
Megvii Engine Team
|
aa953c3bd6
|
fix(mge/module): fix missing import
GitOrigin-RevId: 7cdf6ac814
|
4 years ago |
Megvii Engine Team
|
2e87420865
|
fix(module): fix docs in normalization
GitOrigin-RevId: 326234c35a
|
4 years ago |
Megvii Engine Team
|
7167fdbd49
|
feat(mge/module): add normalization module includes group_norm, instance_norm and layer_norm
GitOrigin-RevId: 9f253e3250
|
4 years ago |
Megvii Engine Team
|
94dba16ff4
|
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
|
4 years ago |
Megvii Engine Team
|
442058aead
|
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
|
4 years ago |
Megvii Engine Team
|
8e11204ab6
|
perf(mge/module): optimize conv_bn qat module to improve performance
GitOrigin-RevId: 9415b83d9c
|
4 years ago |
Megvii Engine Team
|
82b0f67770
|
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
|
4 years ago |
Megvii Engine Team
|
9cb3c07cd4
|
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
|
4 years ago |
Megvii Engine Team
|
8fad00a158
|
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
|
4 years ago |
Megvii Engine Team
|
c89209450d
|
ci(mge): remove blank line for windows doctest
GitOrigin-RevId: 09cf6518e8
|
4 years ago |
Megvii Engine Team
|
b309890c66
|
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
|
4 years ago |
Megvii Engine Team
|
fccb2510c4
|
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
|
4 years ago |
Megvii Engine Team
|
6701346398
|
docs(mge): fix some docstring format problem
GitOrigin-RevId: cbc5ab04b3
|
4 years ago |
Megvii Engine Team
|
20e93630f2
|
feat(mge): rename batch_norm2d -> batch_norm
GitOrigin-RevId: 253e8564ea
|
4 years ago |
Megvii Engine Team
|
7589bce454
|
feat(mge): do not export F.nn.sync_batch_norm
GitOrigin-RevId: aa05e3aac4
|
4 years ago |
Megvii Engine Team
|
d4e2d27213
|
feat(mge): do not export F.nn.conv_bias_activation
GitOrigin-RevId: e536f87fad
|
4 years ago |
Megvii Engine Team
|
cf0507e1ba
|
feat(mge): remove F.identity
GitOrigin-RevId: 858be627ac
|
4 years ago |
Megvii Engine Team
|
164d9cdf1f
|
feat(mge): do not export F.nn.linear
GitOrigin-RevId: 589964729d
|
4 years ago |
Megvii Engine Team
|
fd802e978f
|
feat(mge): do not export F.nn.embedding
GitOrigin-RevId: eb90506535
|
4 years ago |
Megvii Engine Team
|
205a39b07d
|
feat(mge): remove add_update
GitOrigin-RevId: 3068593eeb
|
4 years ago |
Megvii Engine Team
|
67d33303ad
|
feat(mge): remove fast_tanh
GitOrigin-RevId: 7ad5f7ecce
|
4 years ago |
Megvii Engine Team
|
78d2189441
|
feat(mge): rename comparison operators to their full name
GitOrigin-RevId: e503b038a1
|
4 years ago |
Megvii Engine Team
|
2bd84d6789
|
feat(mge): add adaptive pooling python wrapper
GitOrigin-RevId: 789f1511ec
|
4 years ago |
Megvii Engine Team
|
0e3037105c
|
feat(mge/module): add group_local_conv
GitOrigin-RevId: 0503377bc5
|
4 years ago |
Megvii Engine Team
|
026af62042
|
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
|
4 years ago |