Megvii Engine Team
2293385e93
feat(mge): add conv padding mode
GitOrigin-RevId: 147ced856e
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
3eb0505f9b
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
3 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
65af9cfcdc
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
4 years ago
Megvii Engine Team
dc96f6aa49
fix(mge/quantization): fix quantized concat forward problem
GitOrigin-RevId: dc21b340d1
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
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
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
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
442058aead
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
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
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
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
164d9cdf1f
feat(mge): do not export F.nn.linear
GitOrigin-RevId: 589964729d
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago