Megvii Engine Team
|
30ef28b0fa
|
build(imperative): generate stub files for core ops
GitOrigin-RevId: 7971df5f20
|
2 years ago |
Megvii Engine Team
|
dbd9483993
|
feat(dnn,src,imperative): add groupnorm op
GitOrigin-RevId: de3c3d10e5
|
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
|
194201a774
|
revert: chore(mge/misc): api converage
This reverts commit --
77fd432cb2
0d7126641b
559d205a8c
6d8e7398e4
b72517bfe9
GitOrigin-RevId: 18dc4ce999
|
2 years ago |
Megvii Engine Team
|
dfa999ee12
|
fix(doc): fix warp_affine docstring
GitOrigin-RevId: dcec70c872
|
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
|
034c7787fa
|
fix(mge/functional): support non-float32 input when call isxxx
GitOrigin-RevId: ea8f394958
|
2 years ago |
Megvii Engine Team
|
d3b2b51918
|
feat(mge/opr): add meshgrid opr
GitOrigin-RevId: 6f703295be
|
2 years ago |
Megvii Engine Team
|
eddb0aba2b
|
docs(mge/functional): enhance tensor creation function docstring
GitOrigin-RevId: f7682a3b02
|
2 years ago |
Megvii Engine Team
|
222865799b
|
docs(mge/functional): enhance functional statstic func docstring
GitOrigin-RevId: 1045aecf15
|
2 years ago |
Megvii Engine Team
|
213f40437c
|
chore(mge/functional): add normalize interace to f.nn namespace
GitOrigin-RevId: b42c31ca94
|
2 years ago |
Megvii Engine Team
|
f62f261512
|
docs(mge/functional): enhance functional elemwise docstring
GitOrigin-RevId: 5987ed3b59
|
2 years ago |
Megvii Engine Team
|
d205e7bee4
|
docs(mge/functional): update copy example
GitOrigin-RevId: 59aacfe42b
|
2 years ago |
Megvii Engine Team
|
6e48e59334
|
docs(mge/functional): update copy example
GitOrigin-RevId: 2ea102e0ad
|
2 years ago |
Megvii Engine Team
|
e89ee63112
|
docs(mge/functional): update copy example
GitOrigin-RevId: 4011332ba6
|
2 years ago |
Megvii Engine Team
|
7655d99aa5
|
Merge pull request #478 from thunderstudying:docstring-copy
GitOrigin-RevId: 123f95959d
|
2 years ago |
Megvii Engine Team
|
1dd27b39d6
|
docs(mge): update functional.nn.max_pool2d docstring
GitOrigin-RevId: 1d74cec7cb
|
2 years ago |
Megvii Engine Team
|
6e54a3bf31
|
docs(mge): update MaxPool2d & functional.nn.max_pool2d docstring
GitOrigin-RevId: b46f227b59
|
2 years ago |
Megvii Engine Team
|
923dc2d50c
|
Merge pull request #441 from Seeker98:maxpool_doc
GitOrigin-RevId: c56c268dc8
|
2 years ago |
Megvii Engine Team
|
a8b146d74b
|
Merge pull request #397 from kagome112:master
GitOrigin-RevId: e40b12f76e
|
2 years ago |
Megvii Engine Team
|
5655636c1c
|
chore(mge/metric): funtional.topk_acc (add nn) use functional.metric
GitOrigin-RevId: 77fd432cb2
|
2 years ago |
Megvii Engine Team
|
d5688c3d38
|
chore(mge/functional): deprecate debug_param and prefer config
GitOrigin-RevId: 0d7126641b
|
2 years ago |
Megvii Engine Team
|
9bb2e395ce
|
chore(mge/tensorcache): rename the module making it not public
GitOrigin-RevId: 559d205a8c
|
2 years ago |
Megvii Engine Team
|
98460f583f
|
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
|
2 years ago |
thunder
|
3f5e8e9c09
|
docs(mge/functional): update copy example
|
2 years ago |
Megvii Engine Team
|
ada54ba5a0
|
docs(mge): update AvgPool2d & functional.nn.avg_pool2d example
GitOrigin-RevId: e2859c7f99
|
2 years ago |
Megvii Engine Team
|
d8ea01680c
|
docs(mge): update AvgPool2d & functional.nn.max_pool2d docstring
GitOrigin-RevId: d69c5b024f
|
2 years ago |
Megvii Engine Team
|
01353b2d90
|
Merge pull request #442 from Seeker98:avgpool_doc
GitOrigin-RevId: d048955398
|
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
|
5f86368219
|
Revert "feat(dnn): add elemwise modes"
This reverts commit cb713ddb24 .
GitOrigin-RevId: 02adf025e6
|
2 years ago |
Megvii Engine Team
|
d2a1905ad5
|
Revert "feat(mgb): add cumprod opr"
This reverts commit 3436c3bdaa .
GitOrigin-RevId: 95ab3d1aa7
|
2 years ago |
Megvii Engine Team
|
49e14f87b5
|
feat(mgb): add cumprod opr
GitOrigin-RevId: 3436c3bdaa
|
3 years ago |
Megvii Engine Team
|
87aedc2991
|
feat(dnn): add elemwise modes
GitOrigin-RevId: cb713ddb24
|
3 years ago |
Megvii Engine Team
|
cc31b9db34
|
docs(mge/functional): fix vision docs
GitOrigin-RevId: f03567d8ae
|
2 years ago |
Megvii Engine Team
|
6f7649e935
|
docs(docstring): fix pad docstring
GitOrigin-RevId: 698d2f1c0d
|
2 years ago |
Megvii Engine Team
|
51397bfc08
|
feat(mgb): supports value infer and empty input tensor in ElemwiseMultiType
GitOrigin-RevId: 05577a8bc8
|
3 years ago |
Megvii Engine Team
|
247e2f59a4
|
feat(mgb/dnn): add modes that the output type is bool in elemwise
GitOrigin-RevId: fd0134fca2
|
3 years ago |
Megvii Engine Team
|
18f83a258e
|
docs(mge/functional): fix F.svd docstring
GitOrigin-RevId: b84e5fdc46
|
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
|
da91e650a5
|
refactor(ops/layer_norm): speed up the host speed of layer_norm
GitOrigin-RevId: 6f359b5b29
|
3 years ago |
Megvii Engine Team
|
67cfce9f2f
|
fix(imperative/amp): add is_scalar check in elemwise and concat
GitOrigin-RevId: 61a612e92a
|
3 years ago |
Megvii Engine Team
|
d313f92610
|
fix(imperative/amp): fix format transformation for symbol trans
GitOrigin-RevId: 96cc237c67
|
3 years ago |
Megvii Engine Team
|
261a5bce23
|
feat(imperative/amp): add dimshuffle in set_format for nhwc
GitOrigin-RevId: 5ced9e1a31
|
3 years ago |
Megvii Engine Team
|
3892aa0b6e
|
fix(imperative/amp): fix bn params for nhwc amp
GitOrigin-RevId: 57a3b9d418
|
3 years ago |
Megvii Engine Team
|
c28a875fac
|
fix(imperative/amp): adapt new transformation
GitOrigin-RevId: 6edd577a70
|
3 years ago |
Megvii Engine Team
|
fd41302cc1
|
feat(imperative/amp): add set_format
GitOrigin-RevId: 91de6f49de
|
3 years ago |