Megvii Engine Team
|
411253f852
|
feat(mge): implement warp_affine backward
GitOrigin-RevId: 7d7261cf64
|
3 years ago |
Megvii Engine Team
|
7b4b94fd93
|
fix(imperative): fix the segmentfault when reduce backward
GitOrigin-RevId: 8a3e63d4f5
|
3 years ago |
Megvii Engine Team
|
24c5c19bf0
|
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
|
3 years ago |
Megvii Engine Team
|
4aa79c453b
|
perf(mge): override grad of matmul
GitOrigin-RevId: d9d97e70fe
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
99a85c4079
|
fix(mge): fix advanced indexing grad
GitOrigin-RevId: 8033c9322d
|
3 years ago |
Megvii Engine Team
|
7927e98fd6
|
perf(mge): speed up PixelShuffle
GitOrigin-RevId: 942e755745
|
3 years ago |
Megvii Engine Team
|
2a900a69cb
|
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
|
3 years ago |
Megvii Engine Team
|
1709b3940b
|
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
|
3 years ago |
Megvii Engine Team
|
177001d5e5
|
refactor(dispatch): allow dynamic type creation
GitOrigin-RevId: 27dde05cff
|
4 years ago |
Megvii Engine Team
|
ca00177719
|
perf(dispatch): speed up dispatch system
GitOrigin-RevId: eabbe3e021
|
4 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
4 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
000517c641
|
fix(grad): stop using exception in grad_override
GitOrigin-RevId: 00ae38d48b
|
4 years ago |
Megvii Engine Team
|
f12355f727
|
fix(imperative/grad): fix hardcode dtype in subtensor_grad_rule
GitOrigin-RevId: 50da4af26d
|
4 years ago |
Megvii Engine Team
|
0245594156
|
test(autograd): test jvp emulated by 2nd grad
GitOrigin-RevId: 47114fcd99
|
4 years ago |
Megvii Engine Team
|
4f240ec2d3
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
5 years ago |
Megvii Engine Team
|
c49427d15a
|
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
|
5 years ago |
Megvii Engine Team
|
cf3f58cb9b
|
fix(mge/autodiff): fix segfault when grad is nullptr
GitOrigin-RevId: 6139212bfd
|
5 years ago |
Megvii Engine Team
|
288c2e08b5
|
fix(mge/autodiff): fix expand_dims and grad rule fallback
GitOrigin-RevId: 4aae771222
|
5 years ago |
Megvii Engine Team
|
a892e5d0e4
|
perf(mge): add more specialized grad rules
GitOrigin-RevId: f88809a6d7
|
5 years ago |
Megvii Engine Team
|
9fb5581f1d
|
refactor(mge): add specialized grad rule support
GitOrigin-RevId: 141ff0a24f
|
5 years ago |