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
|
4 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 |
Megvii Engine Team
|
bee305beb2
|
feat(traced_module): add functional trace and CallMethod/Function expr
GitOrigin-RevId: ad2cdc1b61
|
4 years ago |
Megvii Engine Team
|
763c56f3b9
|
feat(imperative): add traced module
GitOrigin-RevId: 28c3503f2e
|
4 years ago |
Megvii Engine Team
|
9279104b11
|
feat(mge): add opdef serialization and apply_module_trace
GitOrigin-RevId: 5b45bded1d
|
3 years ago |
Megvii Engine Team
|
ae87876d34
|
feat(mge): refactor weightscaler
GitOrigin-RevId: 7f874388f7
|
3 years ago |
Megvii Engine Team
|
75eb04c559
|
feat(mge/experimental): add WeightScaler support
GitOrigin-RevId: a1d5b68498
|
4 years ago |
Megvii Engine Team
|
8480302da8
|
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
|
4 years ago |