Megvii Engine Team
|
fa4883389a
|
feat(dnn,imperative): remove the restriction of tensor shape when using uint8 region mask
GitOrigin-RevId: 37d9900697
|
2 years ago |
Megvii Engine Team
|
582dd4ceb8
|
fix(dnn/sfotmax): call cpu dispatch for softmax opr
GitOrigin-RevId: a606e66101
|
2 years ago |
Megvii Engine Team
|
f444d4fe4d
|
feat(dnn,imperative): region restricted conv support groups=1 even if
depthwise
GitOrigin-RevId: 950d2f4889
|
2 years ago |
Megvii Engine Team
|
d14ae856ce
|
fix(imperative): update region restricted conv testcase for bias shape
GitOrigin-RevId: 8bf66e7312
|
2 years ago |
Megvii Engine Team
|
ffb626243b
|
ci(imperative): fix imperative test_ctc_loss test
GitOrigin-RevId: d48dccc30c
|
2 years ago |
Megvii Engine Team
|
c1c1d6d160
|
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
|
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
|
cfc41648a4
|
fix(mge): fix grad of maximum(x, x)
GitOrigin-RevId: e0e2efb71b
|
3 years ago |
Megvii Engine Team
|
7252825ce4
|
fix(functional): broadcast_to supports mutable target shape
GitOrigin-RevId: ff79456d5d
|
3 years ago |
Megvii Engine Team
|
24c5c19bf0
|
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
|
3 years ago |
Megvii Engine Team
|
f96429c031
|
feat(imperative): support empty tensor in roi_align
GitOrigin-RevId: aeb2770401
|
3 years ago |
Megvii Engine Team
|
07bdb3bf1e
|
feat(imperative): add swapaxes
GitOrigin-RevId: e84014a011
|
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
|
8563f51404
|
fix(imperative): fix buildin reduce keepdim
GitOrigin-RevId: 38d90ab38a
|
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
|
a926878c01
|
feat(imperative): remove symbolvar of imperative
GitOrigin-RevId: 16da6d1491
|
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
|
90cedd3c1d
|
fix(imperative): restrict using convert_inputs in py_apply
GitOrigin-RevId: b021aac8a6
|
3 years ago |
Megvii Engine Team
|
9779bc7f6d
|
fix(imperative): allow rng op infer shape fallible
GitOrigin-RevId: 687844500c
|
3 years ago |
Megvii Engine Team
|
1709b3940b
|
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
|
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
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
3 years ago |
Megvii Engine Team
|
68cde8734e
|
fix(mge/imperative): support broadcast with None
GitOrigin-RevId: dd330a2a1d
|
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
|
d9a46ea47b
|
fix(dnn): correct behaviour of floor div for int tensor
GitOrigin-RevId: 1444f69cce
|
3 years ago |
Megvii Engine Team
|
2881934cb8
|
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
|
3 years ago |
Megvii Engine Team
|
1f0cc891b0
|
feat(dnn): enable eye to support bool
GitOrigin-RevId: 76d874d5b7
|
3 years ago |
Megvii Engine Team
|
11d75fecb5
|
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
|
3 years ago |
Megvii Engine Team
|
89186edc5d
|
fix(dnn): correct reduce/argmxx/fakequant calculation with nan
GitOrigin-RevId: 7e78bdae91
|
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
|
846cc13ca8
|
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
|
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
|
cebda6ff8d
|
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
|
3 years ago |
Megvii Engine Team
|
f5cb21ed3a
|
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
|
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
|
43fb2c4afb
|
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
|
3 years ago |