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
99309fa3d6
feat(mge/functional): add param output_padding for deconv ops
GitOrigin-RevId: 8a69608953
2 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
60163c5d12
docs(functional): replace elem testcode with doctest format
GitOrigin-RevId: a32a011724
3 years ago
Megvii Engine Team
ab6d12caff
feat(mge): add conv padding mode
GitOrigin-RevId: 147ced856e
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
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
3f3a256e0f
fix(mge/functional): fix conv* dtype promotion
GitOrigin-RevId: 3f03790cfc
3 years ago
Megvii Engine Team
f40df60242
docs(mge): refactor docs to remove warnings
GitOrigin-RevId: efefc2a4a2
3 years ago
Megvii Engine Team
7f9d441cbb
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
4 years ago
Megvii Engine Team
8be70d170c
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
4 years ago
Megvii Engine Team
e18afa0b05
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
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
d970b85deb
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
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
cd02d7c85e
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
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
82847bc70a
docs(mge): add conv2d output shape computation formula
GitOrigin-RevId: 7ee476a80d
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
8fad00a158
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
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
Megvii Engine Team
66f2dbd753
feat(mge/imperative): add module __repr__ for imperative
GitOrigin-RevId: ac13cc4659
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
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
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
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