XindaH
d540d38b8b
Revert "feat(imperative): add pixel_shuffle opr"
This reverts commit d17cd60d3b
.
3 years ago
chenjiahui
d17cd60d3b
feat(imperative): add pixel_shuffle opr
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
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
7234efe1e7
feat(opr): let random op support empty output
GitOrigin-RevId: a2174975aa
3 years ago
Megvii Engine Team
59d5976622
fix(profiler): collect records when thread exit
GitOrigin-RevId: fb796dd61c
4 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
3 years ago
Megvii Engine Team
4688bbabad
refactor(functional): hide functional.vision and replace with functional.nn
GitOrigin-RevId: b6888c7f65
3 years ago
Megvii Engine Team
8796586b89
refactor(functional): import all from metric in nn
GitOrigin-RevId: 41ab78d78d
3 years ago
Megvii Engine Team
27f2ecc436
feat(imperative/opr): add extern c opr and support dump
GitOrigin-RevId: b29e374c60
3 years ago
Megvii Engine Team
91264f3797
fix(traced_module): fix __getattr__ of TracedModuleBuilder
GitOrigin-RevId: 94d91d6938
3 years ago
Megvii Engine Team
526c82c858
feat(traced_module): delete value of node when it will not be used by any expr
GitOrigin-RevId: 3fb7350d01
3 years ago
Megvii Engine Team
6ce1483bc9
fix(utils/network): fix replace var in different networks
GitOrigin-RevId: 8d1356ddb1
3 years ago
Megvii Engine Team
1019865071
feat(module): add tensors and named_tensors
GitOrigin-RevId: cb56d65d38
3 years ago
Megvii Engine Team
bd817f3af2
perf(syncbn): fallback to bn when sync is not required
GitOrigin-RevId: e89fdd87b0
3 years ago
Megvii Engine Team
ffcb4dacd9
style(mge): change the pad opr doc and format
GitOrigin-RevId: 7b20e96519
3 years ago
Megvii Engine Team
43cb305e53
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
3 years ago
Megvii Engine Team
0a665ea452
feat(mge/device): enable to get cuda compute capability
GitOrigin-RevId: b5d3f2225c
3 years ago
Megvii Engine Team
7a9f2ed95e
perf(functional/split): add python binding for Split opr
GitOrigin-RevId: 3359f5da26
3 years ago
Megvii Engine Team
ab309eb5fc
feat(mgb/opr): let Split support empty IO
GitOrigin-RevId: aad6dc06bf
3 years ago
Megvii Engine Team
8485eff12c
docs(mge/functional): add docs for megengine.functional.sub
GitOrigin-RevId: 40208739bd
3 years ago
Megvii Engine Team
e6a8b0256e
fix(mge): ignore errors caused by earlier async errors
GitOrigin-RevId: ce2028d38a
3 years ago
Megvii Engine Team
ee790cb62c
fix(imperative/python): fix warp_perspective doc and arange dtype
GitOrigin-RevId: dc4e021ec0
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
76ce81e828
fix(mge): fix F.nn.dropout train and inference bugs
GitOrigin-RevId: 9d9f246d7b
3 years ago
Megvii Engine Team
5431929ec7
feat(functional): let advance indexing support empty tensor and add more tests
GitOrigin-RevId: 49e1492934
3 years ago
Megvii Engine Team
03e80759b6
feat(mge/distributed): add hybird parallel Opr
GitOrigin-RevId: ff26671746
3 years ago
Megvii Engine Team
e5217db9f9
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
3 years ago
Megvii Engine Team
6d1a4f20e7
feat(traced_module): support tracing submodules in list/dict
GitOrigin-RevId: 4076b47a89
3 years ago
Megvii Engine Team
a3f9073c2c
feat(traced_module): update graph transform and add _module_name
GitOrigin-RevId: ef63ae0fd0
3 years ago
Megvii Engine Team
b3d0affab1
feat(traced_module): support trace custom qat module
GitOrigin-RevId: 49f70a5f46
3 years ago
Megvii Engine Team
15712807b9
feat(traced_module): add name to Node
GitOrigin-RevId: 39c2809067
3 years ago
Megvii Engine Team
e918f0aa75
feat(traced_module): add treedef leaf node check and add some graph api
GitOrigin-RevId: 36c069bfee
3 years ago
Megvii Engine Team
c7e730bc12
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: 09691ebd33
3 years ago
Megvii Engine Team
f88bd3ae33
refactor(traced_module): let TracedModule own argdef_graph_map
GitOrigin-RevId: 80d685b9a3
3 years ago
Megvii Engine Team
b1c46ba46c
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: ac0603057a
3 years ago
Megvii Engine Team
4bb253695b
feat(traced_module): let CallFunction own graph
GitOrigin-RevId: 66cdbca7e5
3 years ago
Megvii Engine Team
9a6a379346
feat(traced_module): add visit method
GitOrigin-RevId: 251ecebf87
4 years ago
Megvii Engine Team
442b4f6c26
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
4 years ago
Megvii Engine Team
f2691566fd
feat(traced_module): add pytree
GitOrigin-RevId: 6c6e53521c
4 years ago