123 Commits (d540d38b8b041b72e04f7235a84c1507d0e18ec5)

Author SHA1 Message Date
  XindaH d540d38b8b Revert "feat(imperative): add pixel_shuffle opr" 3 years ago
  chenjiahui d17cd60d3b 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 43fb2c4afb feat(opr): let roll support empty IO 3 years ago
  Megvii Engine Team b2827cb1b1 feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input 3 years ago
  Megvii Engine Team 50f738772b feat(opr): let Copy and Identity support empty IO 3 years ago
  Megvii Engine Team 8ae61752a0 feat(opr): let nms support empty IO 3 years ago
  Megvii Engine Team 1a1748daf1 feat(opr): let Argsort support empty IO 3 years ago
  Megvii Engine Team 43cb305e53 feat(mge): add imperative pad 3 years ago
  Megvii Engine Team ab309eb5fc feat(mgb/opr): let Split support empty IO 3 years ago
  Megvii Engine Team 76ce81e828 fix(mge): fix F.nn.dropout train and inference bugs 3 years ago
  Megvii Engine Team 5431929ec7 feat(functional): let advance indexing support empty tensor and add more tests 3 years ago
  Megvii Engine Team a430c912fa feat(mgb/opr): let CondTake support empty input 3 years ago
  Megvii Engine Team 03e80759b6 feat(mge/distributed): add hybird parallel Opr 3 years ago
  Megvii Engine Team b74afde89e feat(mgb/opr): let reduce support empty IO 3 years ago
  Megvii Engine Team 3eb0505f9b feat(imperative): add support for quantized conv transpose2d 3 years ago
  Megvii Engine Team a83098890b test(mge/utils): cover all test data 3 years ago
  Megvii Engine Team 0b4a767965 feat(mge/distributed): enable uint8 for collective communication 3 years ago
  Megvii Engine Team f76a2cc2c6 feat(mge/opr): add silu and gelu 3 years ago
  Megvii Engine Team 207a346351 chore(mge): run get_device_count("gpu") in subprocess 4 years ago
  Megvii Engine Team 070c811732 fix(imperative): remove convert_inputs 3 years ago
  Megvii Engine Team 1040b77843 fix(mge/functional): fix F.topk(kth_only=True) 4 years ago
  Megvii Engine Team 1bf18252c4 feat(mge/amp): add mix precision autocast support 4 years ago
  Megvii Engine Team 43098fb8f1 feat(mge): add SlidingWindowTranspose opr 4 years ago
  Megvii Engine Team 7225b0f09f fix(mge/utils): use static infer manager to get value of network.varnode 4 years ago
  Megvii Engine Team 6cd01d5a74 feat(imperative/functional): let elemwise support empty IO & add some tests 4 years ago
  Megvii Engine Team 5198b7834c fix(mge/functional): fix expand_dims for scalar 4 years ago
  Megvii Engine Team 933dd9a497 feat(mge/distributed): add cuda env check before forked thread 4 years ago
  Megvii Engine Team a5060a2bfe feat(mgb/opr): add check_has_inf kernel and opr 4 years ago
  Megvii Engine Team c64b1c94cd feat(imperative/functional): add roll in functional 4 years ago
  Megvii Engine Team d2673c5abf fix(ci/windows): add windows cuda test 4 years ago
  Megvii Engine Team 1525a02530 feat(mge/module): add python wrapper for unfold 4 years ago
  Megvii Engine Team 13b15fb08c feat(megbrain): add correlation opr 4 years ago
  Megvii Engine Team 226f24fa8e test(mge/distributed): add gather scatter reduce broadcast grad test 4 years ago
  Megvii Engine Team b1bf193e4d feat(functional/loss): add reduction choices to loss functions 4 years ago
  Megvii Engine Team 717b88e684 fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode 4 years ago
  Megvii Engine Team 984d85ca26 feat(mge/functional): argmin and argmax support negtive axis 4 years ago
  Megvii Engine Team 6972896933 fix(mge/utils): fix network multiple outputs issue 4 years ago
  Megvii Engine Team 4d9073fcba test(dist): remove redundant assert for reduce_sum and gather 4 years ago
  Megvii Engine Team dee5a10af9 feat(distributed): auto detect device and backend when init group 4 years ago
  Megvii Engine Team 933188aa93 feat(functional/nn): support F.warp_perspective with `mat_idx` 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 9dbe71dd1a feat(mge/utils): add array method for varnode 4 years ago
  Megvii Engine Team 988f897ee2 test(mge): make test stable 4 years ago
  Megvii Engine Team fa1ca0ea6c fix(imperative/opr): fix apply_on_var_node for broadcast 4 years ago
  Megvii Engine Team d970b85deb feat(functional): add conv3d 4 years ago
  Megvii Engine Team 46cad4d3c8 feat(functional/ops): add _assert_equal 4 years ago
  Megvii Engine Team dcfb6a537e refactor(mge/functional): move functional api 4 years ago
  Megvii Engine Team fe9c6e26a8 feat(imperative/opr): deprecate resize op and make it as a special case of interpolate 4 years ago
  Megvii Engine Team fa4bf16800 feat(mge/functional): add repeat and tile opr 4 years ago