Megvii Engine Team
|
669816e291
|
feat(dnn): warpperspective support multi src input
GitOrigin-RevId: 8a4789852e
|
2 years ago |
Megvii Engine Team
|
1a2ed8c47b
|
feat(cuda): add convbias ptx algo testcase
GitOrigin-RevId: 9ad6d4561f
|
2 years ago |
Megvii Engine Team
|
fac67e7c2b
|
feat(gopt): support nchw44 global pooling with fuse_grain
GitOrigin-RevId: 4c43a149f8
|
2 years ago |
Megvii Engine Team
|
5f86368219
|
Revert "feat(dnn): add elemwise modes"
This reverts commit cb713ddb24 .
GitOrigin-RevId: 02adf025e6
|
2 years ago |
Megvii Engine Team
|
d2a1905ad5
|
Revert "feat(mgb): add cumprod opr"
This reverts commit 3436c3bdaa .
GitOrigin-RevId: 95ab3d1aa7
|
2 years ago |
Megvii Engine Team
|
49e14f87b5
|
feat(mgb): add cumprod opr
GitOrigin-RevId: 3436c3bdaa
|
3 years ago |
Megvii Engine Team
|
87aedc2991
|
feat(dnn): add elemwise modes
GitOrigin-RevId: cb713ddb24
|
3 years ago |
Megvii Engine Team
|
c49d3070ba
|
refactor(imperative/ops): extends DnnOprCaller with template
GitOrigin-RevId: 402cba209a
|
3 years ago |
Megvii Engine Team
|
247e2f59a4
|
feat(mgb/dnn): add modes that the output type is bool in elemwise
GitOrigin-RevId: fd0134fca2
|
3 years ago |
Megvii Engine Team
|
99cfefbfe0
|
fix(test): fix test copybara
GitOrigin-RevId: 19b7bdf377
|
3 years ago |
Megvii Engine Team
|
b55942a94d
|
feat(dnn/naive/norm,-dnn/cuda/norm,-dnn/test/norm): add norm dnn opr,
fwd only
GitOrigin-RevId: 989474168d
|
3 years ago |
Megvii Engine Team
|
bbafe69974
|
feat(dnn): add elemwise COND_LT_MOV
GitOrigin-RevId: 444cd6825a
|
3 years ago |
Megvii Engine Team
|
96d90be1c6
|
feat(dnn): fallback support int4 relayout
GitOrigin-RevId: 3625f58470
|
3 years ago |
Megvii Engine Team
|
98b5ee78c1
|
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
|
3 years ago |
Megvii Engine Team
|
5f0e7ffb64
|
feat(fallback): add FB_GI_F32_4x12 benchmark
GitOrigin-RevId: cfacf31b28
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
70209667e8
|
fix(dnn/test): fix some bug when force_deduce_layout is off
GitOrigin-RevId: d7ccc397df
|
3 years ago |
Megvii Engine Team
|
115c4592c0
|
fix(dnn/opencl): fix opencl elemwise tuning issue
GitOrigin-RevId: 317640547d
|
3 years ago |
Megvii Engine Team
|
331567af5d
|
fix(opencl/ci): misc opt and fix:
1: fix megbrain test failed on mali 2.1 devices
2: reduce ci time (about reduce 20min)
GitOrigin-RevId: 4dcdcd48a6
|
3 years ago |
Megvii Engine Team
|
fd6f8e58b0
|
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
|
3 years ago |
Megvii Engine Team
|
8c415f4ed7
|
feat(dnn): cuda nhwc nearest resize support not 1 or 3 channel
GitOrigin-RevId: 764504c341
|
3 years ago |
Megvii Engine Team
|
87de704a46
|
feat(gopt): fuse conv h_swish
GitOrigin-RevId: a3d12991fb
|
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
|
888f4e46ae
|
feat(dnn/cuda): add implicit bmm large kernel dwconv2d dgrad kernels
GitOrigin-RevId: fcb7974d62
|
3 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
3 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
3 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
3 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
3 years ago |
Megvii Engine Team
|
f6d9909460
|
feat(dnn): add elemwise multi type support i16xf32 and u8xf32
GitOrigin-RevId: 2fe469bb4e
|
3 years ago |
kxz@thumt102-1
|
8f48da7ffe
|
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
|
3 years ago |
Megvii Engine Team
|
6bb5409976
|
feat(dnn/src): add images2neibs kernel of opencl and related test
GitOrigin-RevId: 82242b7437
|
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
|
c85631aa77
|
feat(dnn): use ref ptr interface for all backends
GitOrigin-RevId: f65feae5cc
|
3 years ago |
Megvii Engine Team
|
68cdabd288
|
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
|
3 years ago |
Megvii Engine Team
|
056fd6bc59
|
feat(dnn/arm64): support stride_m in arm64 relayout
GitOrigin-RevId: c74193a23d
|
3 years ago |
Megvii Engine Team
|
5885b137fa
|
feat(dnn/arm): support layout like NHWC channel like broadcast on arm
GitOrigin-RevId: fb4300004c
|
3 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
3 years ago |
Megvii Engine Team
|
177dec94c5
|
feat(mgb/opr): add bgr2gray mode for cvtcolor opr
GitOrigin-RevId: d50415b236
|
3 years ago |
Megvii Engine Team
|
682c74df27
|
feat(dnn): add direct nchw88 fp16 conv
GitOrigin-RevId: 44719e8b64
|
3 years ago |
Megvii Engine Team
|
3d3666b6e0
|
test(dnn/bn): add compatible configs for NHWC BN
GitOrigin-RevId: ac757ca307
|
3 years ago |
Megvii Engine Team
|
c33126ab5c
|
feat(mgb/gopt): add reformat manager
GitOrigin-RevId: b9791b131a
|
3 years ago |
Megvii Engine Team
|
8b40f57738
|
feat(mgb/dnn): add conv1x1 algo for matrix mul
GitOrigin-RevId: 585b2c045a
|
3 years ago |
Megvii Engine Team
|
d69b59035d
|
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
|
3 years ago |
Megvii Engine Team
|
bc9cfc277a
|
feat(mgb): add arm resize nchwxx and naive nearest interp
GitOrigin-RevId: d5fbd59a30
|
3 years ago |
Megvii Engine Team
|
722aecd437
|
feat(mgb): support fp16 nhwc backward
GitOrigin-RevId: 954ac6405a
|
3 years ago |
Megvii Engine Team
|
4c13bc7e1b
|
feat(dnn/cuda): add nhwc int8 deconv
GitOrigin-RevId: ad361a0f81
|
3 years ago |
Megvii Engine Team
|
0558b2123d
|
feat(mge/opr): add interpolate nearest mode
GitOrigin-RevId: d384b87f50
|
3 years ago |
Megvii Engine Team
|
c9d060307f
|
feat(dnn/common): add named tensor shape
GitOrigin-RevId: 918928b8ba
|
4 years ago |
Megvii Engine Team
|
336761253d
|
feat(dnn/cuda): add tensorcore matmul for fp16 data type
GitOrigin-RevId: 025c591f75
|
3 years ago |
Megvii Engine Team
|
eab6afab47
|
feat(mgb): add padding opr for megbrain
GitOrigin-RevId: 490e0c5d5a
|
4 years ago |