Megvii Engine Team
|
fe15239ac0
|
fix(imperative): fix error message for tensors with intbx data type
GitOrigin-RevId: cbb42f8127
|
3 years ago |
Megvii Engine Team
|
feb813bcc0
|
chore(release): bump version
GitOrigin-RevId: 1da5b93cae
|
3 years ago |
Megvii Engine Team
|
29f9935d92
|
fix(imperative/python): add layer_norm doc and rm useless param
GitOrigin-RevId: 1b15db621e
|
3 years ago |
Megvii Engine Team
|
300955140e
|
fix(traced_module/doc): fix warning for traced_module docstring
GitOrigin-RevId: 5d2d047d2f
|
3 years ago |
Megvii Engine Team
|
f7d05db7d6
|
fix(mge/module): fix named_tensors
GitOrigin-RevId: bb5aa1f41d
|
3 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
3 years ago |
Megvii Engine Team
|
bfb30dcb81
|
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
|
3 years ago |
Megvii Engine Team
|
2108696588
|
feat(mge/imperative): add local response normalization
GitOrigin-RevId: 939a4d26dd
|
3 years ago |
Megvii Engine Team
|
ca4c93dee7
|
fix(mge): fix layer norm amp bug
GitOrigin-RevId: dba691fcbf
|
3 years ago |
Megvii Engine Team
|
c674bf0e6e
|
feat(debug): add gdb-scripts
GitOrigin-RevId: 43f9674f50
|
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
|
000517c641
|
fix(grad): stop using exception in grad_override
GitOrigin-RevId: 00ae38d48b
|
3 years ago |
Megvii Engine Team
|
d02b0a08f7
|
feat(lite): fix typo
GitOrigin-RevId: 8c46aa3a30
|
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
|
b6142bee9a
|
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
|
3 years ago |
Megvii Engine Team
|
4e95c13617
|
feat(sgd): sgd supports nesterov momentum
GitOrigin-RevId: 13eda179da
|
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
|
270f1aa28f
|
feat(mgb/serialization): add Accessor for OprLoader to fix BN output compatibility
GitOrigin-RevId: 3b95da02c8
|
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
|
b982be567a
|
feat(mge/imperative): add permutation support for the tensor
GitOrigin-RevId: 7ed0447bfe
|
3 years ago |
Megvii Engine Team
|
3977b7aa0b
|
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
|
3 years ago |
Megvii Engine Team
|
884a07ffd2
|
fix(test/random): set a random seed for random unit test
GitOrigin-RevId: ad4b01eac7
|
3 years ago |
Megvii Engine Team
|
3f0bb47a43
|
ci(imperative): add test requires
GitOrigin-RevId: d7a945caf3
|
3 years ago |
Megvii Engine Team
|
11a383bc79
|
feat(mgb): add tensorboard tool c++ layer interface and python parsing
GitOrigin-RevId: ae98bc887f
|
3 years ago |
Megvii Engine Team
|
8084e4e2ee
|
feat(mgb): add tensorboard tool python layer interface
GitOrigin-RevId: 065bc4d153
|
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
|
65d554ed8e
|
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
|
3 years ago |
Megvii Engine Team
|
f642b05e17
|
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
|
3 years ago |
Megvii Engine Team
|
fb20cb36a0
|
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
|
3 years ago |
Megvii Engine Team
|
c7a8d945c7
|
fix(mge/traced_module): let graph record total_id
GitOrigin-RevId: f99178f3ac
|
3 years ago |
Megvii Engine Team
|
4de62ad6cd
|
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
|
3 years ago |
Megvii Engine Team
|
fb49a2834f
|
refactor(mgb/dnn): refactor enum used in serializing
GitOrigin-RevId: e57af4a59c
|
3 years ago |
Megvii Engine Team
|
6082c353e7
|
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
|
4 years ago |
Megvii Engine Team
|
93f70a9539
|
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
|
3 years ago |
Megvii Engine Team
|
b3e958d0bc
|
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
|
3 years ago |
Megvii Engine Team
|
cdb692d2fa
|
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
|
3 years ago |
Megvii Engine Team
|
90dd07161c
|
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
|
3 years ago |
Megvii Engine Team
|
cbf024bf20
|
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
|
3 years ago |
Megvii Engine Team
|
43fb2c4afb
|
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
|
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
|
50f738772b
|
feat(opr): let Copy and Identity support empty IO
GitOrigin-RevId: 4e49d8eae8
|
3 years ago |
Megvii Engine Team
|
74cbc10d82
|
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
|
3 years ago |
Megvii Engine Team
|
8ae61752a0
|
feat(opr): let nms support empty IO
GitOrigin-RevId: 4c51b1aedb
|
3 years ago |
Megvii Engine Team
|
1a1748daf1
|
feat(opr): let Argsort support empty IO
GitOrigin-RevId: 05fcac6e47
|
3 years ago |
Megvii Engine Team
|
7234efe1e7
|
feat(opr): let random op support empty output
GitOrigin-RevId: a2174975aa
|
3 years ago |
Megvii Engine Team
|
3bc94738ba
|
fix(profiler): do not assume clock duration and time_point type
GitOrigin-RevId: 2015bc30bd
|
3 years ago |
Megvii Engine Team
|
b7e6bd7bba
|
feat(profiler): custom event support device
GitOrigin-RevId: 4709c44b39
|
3 years ago |
Megvii Engine Team
|
defbc20efb
|
fix(profiler): fix profiler in dtr
GitOrigin-RevId: 79b5a6b526
|
3 years ago |