Megvii Engine Team
|
9e020d23b4
|
feat(imperative,mgb): add RegionRestrictedConvolution megbrain and imperative opr
GitOrigin-RevId: c0106ade08
|
2 years ago |
Megvii Engine Team
|
411253f852
|
feat(mge): implement warp_affine backward
GitOrigin-RevId: 7d7261cf64
|
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
|
f96429c031
|
feat(imperative): support empty tensor in roi_align
GitOrigin-RevId: aeb2770401
|
3 years ago |
Megvii Engine Team
|
df5ebd3da7
|
fix(imperative/ops): fix the vmemory problem in 1.9
GitOrigin-RevId: 728e983701
|
3 years ago |
Megvii Engine Team
|
673b295d75
|
feat(imperative/amp): remove conv_format and bn param_dim configs
GitOrigin-RevId: 848d34f63d
|
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
|
811579b0e7
|
fix(imperative): restrict value converts to symbolvar
GitOrigin-RevId: 271267be69
|
3 years ago |
Megvii Engine Team
|
9779bc7f6d
|
fix(imperative): allow rng op infer shape fallible
GitOrigin-RevId: 687844500c
|
3 years ago |
Megvii Engine Team
|
27346b0b65
|
test(opr): add scalar check for opr_test
GitOrigin-RevId: dcfd7ad5d6
|
3 years ago |
Megvii Engine Team
|
df3474ca1d
|
perf(functional): rewrite serval elemwise ops with jit subgraph
GitOrigin-RevId: 26247e21d9
|
3 years ago |
Megvii Engine Team
|
d919aaebc7
|
test(imperative): reopen special interpolate test and sync when test rng
GitOrigin-RevId: e3d03b4d1d
|
3 years ago |
Megvii Engine Team
|
5e17b3e4c6
|
Merge pull request #426 from Qsingle:fix-pixel_suffle
GitOrigin-RevId: db9a0f7551
|
3 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
Megvii Engine Team
|
1999307015
|
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
|
3 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
3 years ago |
Qsingle
|
14499e83d9
|
fix the type error of pixel shuffle
add the test of dtype float
|
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
|
7591718d20
|
feat(mge): add functional test
GitOrigin-RevId: aa0be62686
|
3 years ago |
Megvii Engine Team
|
177dec94c5
|
feat(mgb/opr): add bgr2gray mode for cvtcolor opr
GitOrigin-RevId: d50415b236
|
3 years ago |
Megvii Engine Team
|
dc28a43409
|
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
|
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
|
b2827cb1b1
|
feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input
GitOrigin-RevId: 10a3c5b106
|
3 years ago |
Megvii Engine Team
|
8ae61752a0
|
feat(opr): let nms support empty IO
GitOrigin-RevId: 4c51b1aedb
|
3 years ago |
Megvii Engine Team
|
43cb305e53
|
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
|
3 years ago |
Megvii Engine Team
|
76ce81e828
|
fix(mge): fix F.nn.dropout train and inference bugs
GitOrigin-RevId: 9d9f246d7b
|
3 years ago |
Megvii Engine Team
|
5431929ec7
|
feat(functional): let advance indexing support empty tensor and add more tests
GitOrigin-RevId: 49e1492934
|
3 years ago |
Megvii Engine Team
|
a430c912fa
|
feat(mgb/opr): let CondTake support empty input
GitOrigin-RevId: dfb401a945
|
3 years ago |
Megvii Engine Team
|
3eb0505f9b
|
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
|
3 years ago |
Megvii Engine Team
|
a83098890b
|
test(mge/utils): cover all test data
GitOrigin-RevId: e676476b9d
|
3 years ago |
Megvii Engine Team
|
207a346351
|
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
|
4 years ago |
Megvii Engine Team
|
070c811732
|
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
|
3 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
984d85ca26
|
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
|
4 years ago |
Megvii Engine Team
|
933188aa93
|
feat(functional/nn): support F.warp_perspective with `mat_idx`
GitOrigin-RevId: 66910c8bd8
|
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
|
988f897ee2
|
test(mge): make test stable
GitOrigin-RevId: 5de478f8b1
|
4 years ago |
Megvii Engine Team
|
d970b85deb
|
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
|
4 years ago |
Megvii Engine Team
|
46cad4d3c8
|
feat(functional/ops): add _assert_equal
GitOrigin-RevId: b7ce4158b7
|
4 years ago |
Megvii Engine Team
|
dcfb6a537e
|
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
|
4 years ago |
Megvii Engine Team
|
fe9c6e26a8
|
feat(imperative/opr): deprecate resize op and make it as a special case of interpolate
GitOrigin-RevId: a5668c5779
|
4 years ago |
Megvii Engine Team
|
0c37a588ca
|
fix(mge/functional): fix F.ones when input is a tensor of scalar type
GitOrigin-RevId: 6d01d6b58d
|
4 years ago |
Megvii Engine Team
|
cd02d7c85e
|
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
|
4 years ago |