Megvii Engine Team
|
a4d473c99a
|
perf(mge/functional): speed up AddAxis
GitOrigin-RevId: 92a3e1bdd3
|
3 years ago |
Megvii Engine Team
|
3e206d899b
|
perf(mge/functional): speed up Split
GitOrigin-RevId: 43550a0706
|
3 years ago |
Megvii Engine Team
|
729242f9f8
|
refactor(imperative): move typecvt code of sereval ops to c++
GitOrigin-RevId: 4ffaa376c1
|
3 years ago |
Megvii Engine Team
|
3c3fc6f33c
|
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
|
3 years ago |
Megvii Engine Team
|
04193e3bd1
|
feat(dnn): add nearest mode for remap and resize
GitOrigin-RevId: 31e7b72a78
|
3 years ago |
Megvii Engine Team
|
69b89388e8
|
docs(mge/functional): fix debug_param set_execution_strategy docstring
GitOrigin-RevId: 434929c998
|
3 years ago |
Megvii Engine Team
|
90c8a58cca
|
docs(docstring): add pad docstring
GitOrigin-RevId: eaf6a87456
|
3 years ago |
Megvii Engine Team
|
7bd848ce04
|
fix(subgraph): fix hand-written backward for serval jit-elemwise ops
GitOrigin-RevId: ea3a40d96e
|
3 years ago |
Megvii Engine Team
|
21f5a7fcc0
|
fix(subgraph): fix device recognition and scalar propagate
GitOrigin-RevId: fd2fe8bec9
|
3 years ago |
Megvii Engine Team
|
df3474ca1d
|
perf(functional): rewrite serval elemwise ops with jit subgraph
GitOrigin-RevId: 26247e21d9
|
3 years ago |
Megvii Engine Team
|
ca2deebc0f
|
fix(imperative/tensor): make @ operator has the same functionality as matmul functional
GitOrigin-RevId: bf6136cc1a
|
3 years ago |
Megvii Engine Team
|
406115dba0
|
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
|
3 years ago |
Megvii Engine Team
|
bebb2cf4c3
|
Merge pull request #428 from P2Oileen:fix-pad
GitOrigin-RevId: f33ea46ad6
|
3 years ago |
Megvii Engine Team
|
5e17b3e4c6
|
Merge pull request #426 from Qsingle:fix-pixel_suffle
GitOrigin-RevId: db9a0f7551
|
3 years ago |
Megvii Engine Team
|
8e94af9d78
|
Merge pull request #400 from jieli-matrix:docstring-svd
GitOrigin-RevId: 3bcbea3440
|
3 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
3 years ago |
Megvii Engine Team
|
3252016e05
|
Merge pull request #401 from LosReturn:patch-1
GitOrigin-RevId: 440af8bd3d
|
3 years ago |
Megvii Engine Team
|
86ee4638bf
|
Merge pull request #402 from AA1HSHH:docstring-reshape
GitOrigin-RevId: 1ec572eb7c
|
3 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
3 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
Megvii Engine Team
|
d23d1352e7
|
fix(imperative/python): add the default warning for args descending
GitOrigin-RevId: cb5f065e6c
|
3 years ago |
Megvii Engine Team
|
619d78ed86
|
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
|
3 years ago |
Megvii Engine Team
|
fba544881b
|
fix(traced_module): add functional pad to __all__
GitOrigin-RevId: a4fdedfb7f
|
3 years ago |
Megvii Engine Team
|
1999307015
|
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
|
3 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
3 years ago |
jieli-matrix
|
95657d54cf
|
docs(mge/functional): update functional.math.svd docstring
|
3 years ago |
P2Oileen
|
5d755898fd
|
fix(funtional): correct the typo: witdth to width
|
3 years ago |
Megvii Engine Team
|
111fa97514
|
Merge pull request #422 from Asthestarsfalll:zeroslike
GitOrigin-RevId: 4b464446b5
|
3 years ago |
Megvii Engine Team
|
48e0973875
|
docs(mge/functional): update functional.zeros docstring
GitOrigin-RevId: 4ce2752bfc
|
3 years ago |
Qsingle
|
14499e83d9
|
fix the type error of pixel shuffle
add the test of dtype float
|
3 years ago |
Asthestarsfalll
|
520db818d0
|
docs(mge/functional): update functional.zeros_like docstring
|
3 years ago |
Asthestarsfalll
|
b7fe6a27ee
|
docs(mge/functional): update functional.ones_like docstring
|
3 years ago |
Megvii Engine Team
|
2881934cb8
|
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
|
3 years ago |
Megvii Engine Team
|
7cb3ad8a3e
|
fix(mge/functional): fix one_hot irregular coding style
From GITHUB 399
ORIGINAL_AUTHOR=Asthestarsfalll <1186454801@qq.com>
COPYBARA_INTEGRATE_REVIEW=https://github.com/MegEngine/MegEngine/pull/399 from Asthestarsfalll:master 541bf3af29
GITHUB_PUBLIC_PR_NUMBER=399
GITHUB_PR_URL=https://github.com/MegEngine/MegEngine/pull/399
GitOrigin-RevId: 5df007207a
|
3 years ago |
Megvii Engine Team
|
535784d4b7
|
docs(mge/functional): update functional.ones docstring
GitOrigin-RevId: 2b7048f9e6
|
3 years ago |
Megvii Engine Team
|
92012c84e3
|
Merge pull request #398 from ErikQQY:docstring-mul
GitOrigin-RevId: 440dcee883
|
3 years ago |
Megvii Engine Team
|
f398c8e619
|
feat(imperative): add magicmind runtime opr
GitOrigin-RevId: 02ddb886d9
|
3 years ago |
LosReturn
|
2e441ed0fb
|
docs(mge/functional): update functional.clip docstring
|
3 years ago |
DBJ
|
0294303683
|
docs(mge/functional): update functional.tensor.reshape docstring
|
3 years ago |
ErikQQY
|
2d1d5e9b27
|
docs(mge/functional): update functional.mul docstring
|
3 years ago |
Megvii Engine Team
|
3d6fe33e2d
|
Revert "fix(imperative): check async error when getting value"
This reverts commit 3945a9bfa2
GitOrigin-RevId: ec6d3574b2
|
3 years ago |
Megvii Engine Team
|
57e197475b
|
fix(imperative): check async error when getting value
GitOrigin-RevId: 3945a9bfa2
|
3 years ago |
Megvii Engine Team
|
7ac693a882
|
docs(imperative): fix multiple target warning when cross-reference
GitOrigin-RevId: c5b4884c8a
|
3 years ago |
Megvii Engine Team
|
11d75fecb5
|
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
|
3 years ago |
Megvii Engine Team
|
ea3eb66e7f
|
docs(mge): modify some docstring about conv and tools
GitOrigin-RevId: 4833e78d71
|
3 years ago |
Megvii Engine Team
|
acd835d0df
|
docs(mge): add some docs for cvt_color and GraphInference
GitOrigin-RevId: d78628bfcf
|
3 years ago |
Megvii Engine Team
|
5c94db16f2
|
feat(mge/functional): add groups support for conv_transpose2d & 3d
GitOrigin-RevId: b75b792fb4
|
3 years ago |
Megvii Engine Team
|
8a222d2c9c
|
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
|
3 years ago |
Megvii Engine Team
|
27aa648ba7
|
fix(mge): control compute_mode by context
GitOrigin-RevId: f3968309f4
|
3 years ago |
Megvii Engine Team
|
846cc13ca8
|
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
|
3 years ago |