Megvii Engine Team
|
b2ed2630f7
|
Revert "fix(mge/functional): rm useless arguments"
This reverts commit 8e4f25bfd8 .
GitOrigin-RevId: df696ab8a2
|
5 years ago |
Megvii Engine Team
|
d23fec06cb
|
fix(mge/functional): rm useless arguments
group and compute_mode are not used by local_conv2d
GitOrigin-RevId: 8e4f25bfd8
|
5 years ago |
Megvii Engine Team
|
4500faf1ea
|
fix(python_module/mge): one_hot: no default value for num_classes
GitOrigin-RevId: c4a5310880
|
5 years ago |
Megvii Engine Team
|
8c110c3942
|
feat(mge/quantization): add quantization interface
GitOrigin-RevId: 4fe1233ec3
|
5 years ago |
Megvii Engine Team
|
814a427810
|
feat(mge/module): add `with_parent` argument in `_flatten`
GitOrigin-RevId: 1c88559ece
|
5 years ago |
Megvii Engine Team
|
d3730036ea
|
test(mge/dist): add unit tests for megengine.distributed.utils
GitOrigin-RevId: 32336c66f1
|
5 years ago |
Megvii Engine Team
|
80d9e56b38
|
fix(mge/tensor): where support x or y is empty
GitOrigin-RevId: 2b69ee2e79
|
5 years ago |
Megvii Engine Team
|
4c5d01fa34
|
refactor(mge/examples): add example to use modified sublinear API
GitOrigin-RevId: c6f6c65d0b
|
5 years ago |
Megvii Engine Team
|
396653df5f
|
feat(mge/api): expose sublinear related parameters at mge api level
GitOrigin-RevId: 7a47f5d0d5
|
5 years ago |
Megvii Engine Team
|
dd962a4700
|
fix(mge/module): add negative slope attribute for LeakyReLU module
GitOrigin-RevId: 67e1967b56
|
5 years ago |
Megvii Engine Team
|
5507a29bd3
|
feat(mge/functional): add normalize opr
GitOrigin-RevId: 572a32f563
|
5 years ago |
Megvii Engine Team
|
000663c30c
|
feat(mge/functional): add isnan and isinf oprs
GitOrigin-RevId: b4a347751c
|
5 years ago |
Megvii Engine Team
|
9b41321926
|
fix(mge/functional): support negative axis in math.py
GitOrigin-RevId: 75143a7309
|
5 years ago |
Megvii Engine Team
|
207527d108
|
fix(mge/clamp): fix `F.clamp`
GitOrigin-RevId: 1efac8add6
|
5 years ago |
Megvii Engine Team
|
bb92ee26e3
|
fix(mge/functional): refine doc and add test for linspace
GitOrigin-RevId: d6c7eea2c2
|
5 years ago |
Megvii Engine Team
|
ce30d045ee
|
feat(mge/functional): add arange in tensor.py
GitOrigin-RevId: ad88a4c18e
|
5 years ago |
Megvii Engine Team
|
fc4dedc0bc
|
fix(mge/random): add lower bound and higher bound for uniform sampling
GitOrigin-RevId: 2a2c56fd17
|
5 years ago |
Megvii Engine Team
|
712b87c8c1
|
feat(mgb/core): add comp node for cambricon
add testcase for cambricon comp node
GitOrigin-RevId: 7794faa47f
|
5 years ago |
Megvii Engine Team
|
d74a4ee931
|
fix(mge/build): use consistent genfiles/megbrain_build_config.h
Modify get_dynamic_info to work with cmake changes
Also rename get_dynamic_info as get_mem_allocation_info
GitOrigin-RevId: 383affe388
|
5 years ago |
Megvii Engine Team
|
e8494f729b
|
fix(unittest): remove the torch dependency in test_conv
GitOrigin-RevId: 36dcab2b95
|
5 years ago |
Megvii Engine Team
|
7ed98dd50a
|
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
|
5 years ago |
Megvii Engine Team
|
044695aa2a
|
feat(mgb/plugin): add profiler for dynamic graph
GitOrigin-RevId: 10a6cccff9
|
5 years ago |
Megvii Engine Team
|
62b0c6cd27
|
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
|
5 years ago |
Megvii Engine Team
|
dd41e2e47a
|
fix(mge/test):fix integration/manual tests
GitOrigin-RevId: 5933392407
|
5 years ago |
Megvii Engine Team
|
8d2bbf7383
|
fix(mge/functional): support scalar inputs in elemwise functions
GitOrigin-RevId: 7bce561ee1
|
5 years ago |
Megvii Engine Team
|
65432d3b93
|
fix(mge/module): fix torch subgraph under jit.trace with symbolic=False
GitOrigin-RevId: a208ba79d9
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |