Megvii Engine Team
d502e79f85
feat(mge): make F.norm numpy compatible
GitOrigin-RevId: 4e5236171c
4 years ago
Megvii Engine Team
1e2117f66d
feat(mge): remove add_axis and remove_axis
GitOrigin-RevId: 59611d43f9
4 years ago
Megvii Engine Team
e034874ae7
feat(mge): rename clamp -> clip
GitOrigin-RevId: 2b594a611e
4 years ago
Megvii Engine Team
20cdd11d4d
docs(mge): fix docstring of cgtools
GitOrigin-RevId: ee8c93b9f7
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
f4860b9345
fix(mge/functional): fix trace topk
GitOrigin-RevId: c88ca8219b
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
7e8f720953
refactor(mge/tensor): tensor reduce supports keepdims
GitOrigin-RevId: 8ed95e0fb8
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago