Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
4 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
4 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
4 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
4 years ago |
kxz@thumt102-1
|
8f48da7ffe
|
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
|
4 years ago |
Megvii Engine Team
|
b8ccc6a211
|
fix(mgb): fix loss execution policy after opr shallow copy
GitOrigin-RevId: 4738136e4a
|
4 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
dc28a43409
|
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
|
4 years ago |
Megvii Engine Team
|
270f1aa28f
|
feat(mgb/serialization): add Accessor for OprLoader to fix BN output compatibility
GitOrigin-RevId: 3b95da02c8
|
4 years ago |
Megvii Engine Team
|
c0ccd0ea7e
|
feat(mge/bn): add NHWC support for bn
GitOrigin-RevId: 0a5bb6f72d
|
4 years ago |
Megvii Engine Team
|
74cbc10d82
|
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
|
4 years ago |
Megvii Engine Team
|
3eb0505f9b
|
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
|
4 years ago |
Megvii Engine Team
|
bba04f02e5
|
feat(mgb/gopt): add fusion support for conv, astype(s4) and reformat
GitOrigin-RevId: 6329ca2c5f
|
4 years ago |
Megvii Engine Team
|
e6caa9ff89
|
feat(opr): add bn backward for inference mode
GitOrigin-RevId: bb643cb62f
|
4 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
Megvii Engine Team
|
f30c0e06a6
|
feat(mgb/opr): add lsq opr
GitOrigin-RevId: 45494a2b57
|
4 years ago |
Megvii Engine Team
|
1cfdbc565c
|
feat(dnn): add deterministic max pooling
GitOrigin-RevId: 9ab4c7a748
|
4 years ago |
Megvii Engine Team
|
f584416aa2
|
fix(dnn/bn): revise the conditions for inplace flag
GitOrigin-RevId: 59a104bf6a
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
8494a1529e
|
chore(scripts): clarify and fix default value of bit combined enum
GitOrigin-RevId: 3716bf9bb5
|
4 years ago |
Megvii Engine Team
|
55974e8cf9
|
feat(log): opt log
* opt log at release mode
* add MGE_OVERRIDE_LOG_LEVEL for runtime debug
//! env to config LogLevel
//! DEBUG = 0, INFO = 1, WARN = 2, ERROR = 3, NO_LOG = 4
//! for example , export MGE_OVERRIDE_LOG_LEVEL=0, means set LogLevel to DEBUG
GitOrigin-RevId: 16cd674c56
|
4 years ago |
Megvii Engine Team
|
ba2ad46e54
|
feat(gopt): add deconv nchw4 int8 opt pass, add deconv nchw int8
GitOrigin-RevId: c0530a949e
|
4 years ago |
Megvii Engine Team
|
a3ea1f153c
|
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
|
5 years ago |
Megvii Engine Team
|
3fab21ed31
|
fix(lite): fix BUILD and misc
GitOrigin-RevId: 48aa0e0924
|
4 years ago |
Megvii Engine Team
|
97207d00b0
|
fix(mgb/opr): fix bazel compile error by comma in mgb_throw_if's format string
GitOrigin-RevId: 72d19aecd1
|
5 years ago |
Megvii Engine Team
|
b1806b8432
|
docs(mgb/opr): improve batch_norm err msg in training mode
GitOrigin-RevId: 4c43bc8b0b
|
5 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
5 years ago |
Megvii Engine Team
|
5db17907b4
|
chore(megbrain): fix megskull cnn/img opr compatiblity
for add convolutionv2 param
GitOrigin-RevId: 9d0cf55ef4
|
5 years ago |
Megvii Engine Team
|
935fe7819e
|
chore(megbrain): format sereg.h code
GitOrigin-RevId: 0fe1cf6a8b
|
5 years ago |
Megvii Engine Team
|
55042195d4
|
chore(winograd): add Convolutionv2 param
GitOrigin-RevId: 1a9e2ea340
|
5 years ago |
Megvii Engine Team
|
5063a2063b
|
feat(mgb): add fastrun support for matmul/batchedmatrixmul
GitOrigin-RevId: a48ea9bff6
|
5 years ago |
Megvii Engine Team
|
d07cfdcbe9
|
refactor(mgb): move convolution mixin to search_policy
GitOrigin-RevId: 81e32da034
|
5 years ago |
Megvii Engine Team
|
a57317f4e1
|
refactor(mgb): move profile cache out of mgb opr and update CACHE_KEY_VERSION
GitOrigin-RevId: db433164d2
|
5 years ago |
Megvii Engine Team
|
a85531dd0f
|
feat(mgb/opr): add tqt opr
GitOrigin-RevId: 49c62cd532
|
5 years ago |
Megvii Engine Team
|
f470df4f1a
|
fix(mgb/opr): fix convbias with no bias when weight preprocess
GitOrigin-RevId: 8837ad539c
|
5 years ago |
Megvii Engine Team
|
60c6d59fc9
|
feat(mbg/core): support bias preprocess in conv_bias
GitOrigin-RevId: d2e1e14d41
|
5 years ago |
Megvii Engine Team
|
b60cc8ca3e
|
feat(mgb/opr): add megbrain fake quant opr
GitOrigin-RevId: a858bce939
|
5 years ago |
Megvii Engine Team
|
9415ba5890
|
feat(src/core): free weight preprocessed weight
GitOrigin-RevId: 5f91acb909
|
5 years ago |
Megvii Engine Team
|
1e90c45793
|
refactor(mgb/opr): move fastrun out of conv opr
GitOrigin-RevId: d5ef5356f6
|
5 years ago |
Megvii Engine Team
|
2a3f4d099a
|
refactor(dnn/arm): refactor CPU heuristic algo selection
GitOrigin-RevId: 60d2646bb3
|
5 years ago |
Megvii Engine Team
|
ca717806af
|
fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD
GitOrigin-RevId: e66dabee01
|
5 years ago |
Megvii Engine Team
|
8d1f302590
|
fix(mge): fix batch norm dump
GitOrigin-RevId: eb739437ef
|
5 years ago |
Megvii Engine Team
|
5a85c907e0
|
feat(mgb/opr): add megbrain adaptive pooling opr
GitOrigin-RevId: 82833f41d9
|
5 years ago |
Megvii Engine Team
|
fe5649e456
|
fix(mge/imperative): remove duplicated opr
GitOrigin-RevId: 7d49785fad
|
5 years ago |
Megvii Engine Team
|
add3a1bc62
|
feat(mgb/opr): add weight preprocess option
GitOrigin-RevId: 9d83a174fa
|
5 years ago |
Megvii Engine Team
|
94f1f63496
|
refactor(mgb/build): remove MGB_ENABLE_IMPERATIVE macro
GitOrigin-RevId: 94a40798ab
|
5 years ago |
Megvii Engine Team
|
354b478c1f
|
fix(opr): remove constant value infer for const SharedDeviceTensor
GitOrigin-RevId: 8fa47b35ad
|
5 years ago |
Megvii Engine Team
|
76c79d79a1
|
fix(mgb/opr): fix ConvBias not passing on prep_filter
GitOrigin-RevId: 0dc9f9d133
|
5 years ago |
Megvii Engine Team
|
498f7b2ea7
|
fix(opr): fix compile on GCC6
GitOrigin-RevId: b59756c923
|
5 years ago |
Megvii Engine Team
|
b9f3077d26
|
feat(opr): remove old weight preprocess adapter in fastrun
GitOrigin-RevId: f467c0db80
|
5 years ago |