170 Commits (dbd94839937de569455f74da819567db28dc5f8d)

Author SHA1 Message Date
  Megvii Engine Team 90c8a58cca docs(docstring): add pad docstring 3 years ago
  Megvii Engine Team 7bd848ce04 fix(subgraph): fix hand-written backward for serval jit-elemwise ops 3 years ago
  Megvii Engine Team 21f5a7fcc0 fix(subgraph): fix device recognition and scalar propagate 3 years ago
  Megvii Engine Team df3474ca1d perf(functional): rewrite serval elemwise ops with jit subgraph 3 years ago
  Megvii Engine Team 406115dba0 fix(imperative): syncbn fp16 support 3 years ago
  Megvii Engine Team bebb2cf4c3 Merge pull request #428 from P2Oileen:fix-pad 3 years ago
  Megvii Engine Team 5e17b3e4c6 Merge pull request #426 from Qsingle:fix-pixel_suffle 3 years ago
  Megvii Engine Team cbbca5fb10 feat(mge): add softmax op use cudnn api 3 years ago
  Megvii Engine Team 0bdd0b1467 refactor(dispatch): switch to new dispatch system 3 years ago
  Megvii Engine Team fba544881b fix(traced_module): add functional pad to __all__ 3 years ago
  Megvii Engine Team 1999307015 feat(mgb/opr): add dropout kernel 3 years ago
  Megvii Engine Team a93741815b feat(mgb/opr): add layernorm forward and backward kernel 3 years ago
  P2Oileen 5d755898fd fix(funtional): correct the typo: witdth to width 3 years ago
  Qsingle 14499e83d9 fix the type error of pixel shuffle 3 years ago
  Megvii Engine Team 7cb3ad8a3e fix(mge/functional): fix one_hot irregular coding style 3 years ago
  “wenjuan” 96ffcb2dc5 docs(mge/functional): update functional.batch_norm docstring 3 years ago
  Megvii Engine Team 7ac693a882 docs(imperative): fix multiple target warning when cross-reference 3 years ago
  Megvii Engine Team ea3eb66e7f docs(mge): modify some docstring about conv and tools 3 years ago
  Megvii Engine Team 5c94db16f2 feat(mge/functional): add groups support for conv_transpose2d & 3d 3 years ago
  Megvii Engine Team 8a222d2c9c fix(mge): replace functional paramter 3 years ago
  Megvii Engine Team 29f9935d92 fix(imperative/python): add layer_norm doc and rm useless param 3 years ago
  Megvii Engine Team 2108696588 feat(mge/imperative): add local response normalization 3 years ago
  Megvii Engine Team ca4c93dee7 fix(mge): fix layer norm amp bug 3 years ago
  Megvii Engine Team c0ccd0ea7e feat(mge/bn): add NHWC support for bn 3 years ago
  chenjiahui 6b863cc53a feat(imperative): add pixel_shuffle opr 3 years ago
  Megvii Engine Team 97b1b7774d feat(mgb): add megbrain layer norm opr with subgraph 3 years ago
  Megvii Engine Team 4688bbabad refactor(functional): hide functional.vision and replace with functional.nn 3 years ago
  Megvii Engine Team 8796586b89 refactor(functional): import all from metric in nn 3 years ago
  Megvii Engine Team bd817f3af2 perf(syncbn): fallback to bn when sync is not required 3 years ago
  Megvii Engine Team ffcb4dacd9 style(mge): change the pad opr doc and format 3 years ago
  Megvii Engine Team 43cb305e53 feat(mge): add imperative pad 3 years ago
  Megvii Engine Team c5f8b58385 docs(docstring): transfer to google style 3 years ago
  Megvii Engine Team 76ce81e828 fix(mge): fix F.nn.dropout train and inference bugs 3 years ago
  Megvii Engine Team d063d5774f perf(functional): use fma to reduce elemwise but disable subgraph compilation 3 years ago
  Megvii Engine Team 2a063f8e87 fix(subgraph): fix scope mismatch of subgraph content 3 years ago
  Megvii Engine Team 8c47c1f149 perf(syncbn): reimplement with subgraph 3 years ago
  Megvii Engine Team 77309609fa perf(functional/dropout): add fastpath for dropout 3 years ago
  Megvii Engine Team 3f3a256e0f fix(mge/functional): fix conv* dtype promotion 3 years ago
  Megvii Engine Team f76a2cc2c6 feat(mge/opr): add silu and gelu 3 years ago
  Megvii Engine Team 070c811732 fix(imperative): remove convert_inputs 3 years ago
  Megvii Engine Team 5cef74a77e feat(mge/amp): add GradScaler support 4 years ago
  Megvii Engine Team 1bf18252c4 feat(mge/amp): add mix precision autocast support 4 years ago
  Megvii Engine Team 75eb04c559 feat(mge/experimental): add WeightScaler support 4 years ago
  Megvii Engine Team 43098fb8f1 feat(mge): add SlidingWindowTranspose opr 4 years ago
  Megvii Engine Team 5e54821845 docs(syncbn): complete syncbn document 4 years ago
  Megvii Engine Team 1525a02530 feat(mge/module): add python wrapper for unfold 4 years ago
  Megvii Engine Team 7f9d441cbb docs(mge/module): add note about weight shape in conv 4 years ago
  Megvii Engine Team e18afa0b05 feat(mge/module): python wrapper for conv_transpose3d 4 years ago
  Megvii Engine Team 65af9cfcdc refactor(mge): use lower case for default string parameters in functional and module 4 years ago
  Megvii Engine Team d0aa9b41ee refactor(mge/functional): move nvof to vision, compatible with old usage 4 years ago