Megvii Engine Team
0c37a588ca
fix(mge/functional): fix F.ones when input is a tensor of scalar type
GitOrigin-RevId: 6d01d6b58d
4 years ago
Megvii Engine Team
cd02d7c85e
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
4 years ago
Megvii Engine Team
4485e780ae
fix(mge/functional): fix linspace device and open other trace tests
GitOrigin-RevId: 4667c4adec
4 years ago
Megvii Engine Team
334eda8717
refactor(mge): test trace inside opr_test
GitOrigin-RevId: 2cf1135c1c
4 years ago
Megvii Engine Team
194660462c
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
4 years ago
Megvii Engine Team
e3a3e0cd3c
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
4aaae9950b
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
4 years ago
Megvii Engine Team
df976782fa
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
4 years ago
Megvii Engine Team
34c705fcbf
refactor(mge/imperative): move detach into C++
GitOrigin-RevId: 8c0d86cbbf
4 years ago
Megvii Engine Team
442058aead
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
4 years ago
Megvii Engine Team
8fad00a158
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
4 years ago
Megvii Engine Team
28dbadf729
feat(imperative): add remap opr
GitOrigin-RevId: 1fa7e18d4b
4 years ago
Megvii Engine Team
215f88f373
fix(dnn/argmxx): fix argmxx on inf
GitOrigin-RevId: 740f67b73a
4 years ago
Megvii Engine Team
1fa143ce87
refactor(mge/functional): matmul supports symbolic shape, batched mv multiply
GitOrigin-RevId: c4d8cf3306
4 years ago
Megvii Engine Team
9748aebeea
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
4 years ago
Megvii Engine Team
d4e2d27213
feat(mge): do not export F.nn.conv_bias_activation
GitOrigin-RevId: e536f87fad
4 years ago
Megvii Engine Team
7d8a7e3efe
feat(mge): do not export F.loss.*
GitOrigin-RevId: a3ce4d3d76
4 years ago
Megvii Engine Team
5b2f0129cb
feat(mge): rename cross_entropy_with_softmax -> cross_entropy
GitOrigin-RevId: 9435c3260a
4 years ago
Megvii Engine Team
5b7ae26849
feat(mge): do not export F.nn.roi_pooling
GitOrigin-RevId: 8a07dd1da4
4 years ago
Megvii Engine Team
d22d1676e7
feat(mge): do not export F.nn.roi_align
GitOrigin-RevId: c2ac7bab32
4 years ago
Megvii Engine Team
9c5611345f
feat(mge): do not export F.nn.nms
GitOrigin-RevId: 7a0a328c37
4 years ago
Megvii Engine Team
e42679b517
feat(mge): do not export F.nn.interpolate
GitOrigin-RevId: 031c6555c0
4 years ago
Megvii Engine Team
a640eeff62
feat(mge): remove batched_nms
GitOrigin-RevId: 01f9ee137c
4 years ago
Megvii Engine Team
205a39b07d
feat(mge): remove add_update
GitOrigin-RevId: 3068593eeb
4 years ago
Megvii Engine Team
e0da74852e
feat(opr): add confidential operator
GitOrigin-RevId: 53c2d4bc45
4 years ago
Megvii Engine Team
2bd84d6789
feat(mge): add adaptive pooling python wrapper
GitOrigin-RevId: 789f1511ec
4 years ago
Megvii Engine Team
6ce796aec4
test(mge): replace `assertTensorClose` with `np.testing.assert_allclose`
GitOrigin-RevId: 2bf8c42cfd
4 years ago
Megvii Engine Team
b80fade3f9
fix(mge/core): avoid create RawTensor with zero-stride numpy ndarray
GitOrigin-RevId: a9b2940bdc
4 years ago
Megvii Engine Team
bd7f885a2b
fix(mge/pytest): remove __init__ in py-test
GitOrigin-RevId: e16eae9cbc
4 years ago
Megvii Engine Team
b970c6daf1
fix(functional): hide format interface and skip test_conv_bias
GitOrigin-RevId: 7de75bf084
4 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
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
939bf3b28d
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
4 years ago
Megvii Engine Team
03320a0548
fix(functional): change to user-friendly nms api and fix concat
GitOrigin-RevId: 76b4190270
4 years ago
Megvii Engine Team
7fadc16d3c
refactor(mge/functional): support tensor shape in interpolate and split
GitOrigin-RevId: 6430b64f01
4 years ago
Megvii Engine Team
40d18c8920
fix(mge/imperative): fix tests when shape is tensor
GitOrigin-RevId: fd0095c1ec
4 years ago
Megvii Engine Team
f4927db2fb
feat(mge/functional): support where func
GitOrigin-RevId: 9df6421ebe
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