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
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
213f40437c
chore(mge/functional): add normalize interace to f.nn namespace
GitOrigin-RevId: b42c31ca94
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
98460f583f
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
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
87aedc2991
feat(dnn): add elemwise modes
GitOrigin-RevId: cb713ddb24
3 years ago
Megvii Engine Team
6f7649e935
docs(docstring): fix pad docstring
GitOrigin-RevId: 698d2f1c0d
2 years ago
Megvii Engine Team
24c5c19bf0
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
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
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
Megvii Engine Team
fc633ce4ff
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
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
8182af6eb6
fix(mgb): fix strategy of grad_op and opr_attr
GitOrigin-RevId: bb7ab8fa9d
3 years ago
Megvii Engine Team
409c988163
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
3 years ago
Seeker
275a492be6
docs(mge/functional): update functional.nn.max_pool2d docstring
3 years ago
Seeker98
a22669c132
docs(mge/functional): update functional.nn.avg_pool2d docstring
3 years ago
Megvii Engine Team
7927e98fd6
perf(mge): speed up PixelShuffle
GitOrigin-RevId: 942e755745
3 years ago
Megvii Engine Team
f7e10ea8d7
perf(imperative): improve matmul/batch_matmul
GitOrigin-RevId: 4ceb2eb601
3 years ago
Megvii Engine Team
c2435d1561
perf(imperative): specialize adaptive pooling
GitOrigin-RevId: 01e1418458
3 years ago
Megvii Engine Team
d2278f02d2
perf(imperative): speed up conv_transpose3d
GitOrigin-RevId: e741305446
3 years ago
Megvii Engine Team
3f11b4213a
docs(functional): replace megengine function testcode with doctest format
GitOrigin-RevId: f91d5f5afd
3 years ago
Megvii Engine Team
60163c5d12
docs(functional): replace elem testcode with doctest format
GitOrigin-RevId: a32a011724
3 years ago
Megvii Engine Team
2d72de8a87
fix(mge): fix infer output attrs fallible
GitOrigin-RevId: ea18c7f753
3 years ago
Megvii Engine Team
22be7e11ca
docs(mge/functional): typo
GitOrigin-RevId: 0ebc6cb322
3 years ago
Megvii Engine Team
111c150db1
docs(mge/functional): fix typo
GitOrigin-RevId: 9473f17970
3 years ago
Megvii Engine Team
273c0e8745
fix(autodiff): fix some bugs in relation to 2nd order grad
1. implement double backward for batchnorm
2. fix grad attach in nested grad manager
3. pad empty tensor for unsatisfied output_has_grad
4. support double backward for jit subgraph
5. support double backward for autodiff.Function
6. readd debug flag MGE_LOG_OP_DISPATCH
GitOrigin-RevId: cd31ddc620
3 years ago
Megvii Engine Team
a4327c4d25
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
3 years ago
Megvii Engine Team
fa62f6c06e
perf(mge/utils): move convert_input into C++
GitOrigin-RevId: 0d1cd36251
3 years ago
Megvii Engine Team
d98be08030
perf(mge): move Const into C++
GitOrigin-RevId: 31a443cffd
3 years ago
Megvii Engine Team
729242f9f8
refactor(imperative): move typecvt code of sereval ops to c++
GitOrigin-RevId: 4ffaa376c1
3 years ago
Megvii Engine Team
3c3fc6f33c
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
3 years ago