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 |
Megvii Engine Team
|
1af350c6d2
|
feat(dnn): add fill kernel
GitOrigin-RevId: d2cee3a7a0
|
3 years ago |
Megvii Engine Team
|
3eb0505f9b
|
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
|
3 years ago |
Megvii Engine Team
|
ddba5c9674
|
fix(core): fix nr_threads is zero
GitOrigin-RevId: 0ccbe3c69b
|
3 years ago |
Megvii Engine Team
|
2aba0378b9
|
refactor(mgb/dnn): fix group conv is_available
GitOrigin-RevId: b279909168
|
3 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
Megvii Engine Team
|
83e4c9d7ab
|
fix(opencl): open opencl topk test when opencl beyond 2.0
GitOrigin-RevId: f2ad6b4af2
|
4 years ago |
Megvii Engine Team
|
f30c0e06a6
|
feat(mgb/opr): add lsq opr
GitOrigin-RevId: 45494a2b57
|
4 years ago |
Megvii Engine Team
|
1cfdbc565c
|
feat(dnn): add deterministic max pooling
GitOrigin-RevId: 9ab4c7a748
|
4 years ago |
Megvii Engine Team
|
a5060a2bfe
|
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
|
4 years ago |
Megvii Engine Team
|
ca2828ddcb
|
fix(dnn/x86): fix x86 int8 matmul ldc bug
GitOrigin-RevId: 2502f99000
|
4 years ago |
Megvii Engine Team
|
86b69cacd0
|
fix(dnn): fixes for int4
GitOrigin-RevId: 845e164fd3
|
4 years ago |
Megvii Engine Team
|
4a802d21ca
|
feat(dnn/cuda): add conv u4xs4 sass kernel
GitOrigin-RevId: 4defcf5f1f
|
4 years ago |
Megvii Engine Team
|
56e863b7d4
|
fix(dnn/cuda): fix int4 epilogue stg bug
GitOrigin-RevId: e86da9a8a8
|
4 years ago |
Megvii Engine Team
|
df1af59b5c
|
feat(dnn): warp perspective support int4
GitOrigin-RevId: 826a43b349
|
4 years ago |
Megvii Engine Team
|
2398df079c
|
feat(dnn/cuda): add cuda int4 pooling
GitOrigin-RevId: 14ed4e6f00
|
4 years ago |
Megvii Engine Team
|
e250afb08f
|
feat(dnn/cuda): support conv_bias for nchw64 and qint4
GitOrigin-RevId: 1c65ba87d7
|
4 years ago |
Megvii Engine Team
|
3b9b87809d
|
refactor(dnn): refactor lowbit tensor format
GitOrigin-RevId: b646dc085b
|
4 years ago |
Megvii Engine Team
|
8fef78d06d
|
feat(dnn/cuda): add relayout format when width is an odd number
GitOrigin-RevId: f059f1f56d
|
4 years ago |
Megvii Engine Team
|
91d6160769
|
feat(dnn/common): add tensor format for low-bits tensor layout
GitOrigin-RevId: 0aa3753f37
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
1997b1a289
|
feat(dnn/cuda): add correlation kernel
GitOrigin-RevId: 25e58b61e6
|
4 years ago |
Megvii Engine Team
|
c3f8cf04fa
|
feat(dnn): add conv_bwd_data and conv_bwd_filter accuracy shake check
GitOrigin-RevId: 4069e083d2
|
4 years ago |
Megvii Engine Team
|
1e6ef3771f
|
feat(mgb/dnn): add accuracy shake checker
GitOrigin-RevId: 0bb52078a1
|
4 years ago |
Megvii Engine Team
|
ff755451d2
|
refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method
GitOrigin-RevId: 144ff547d1
|
4 years ago |
Megvii Engine Team
|
a49f4a66b7
|
feat(dnn): add indexing_one_hot and indexing_set_one_hot opr
GitOrigin-RevId: c5406c71ff
|
4 years ago |