14 Commits (75c413e3bb0b58881cc46faa2647af48f79c29e3)

Author SHA1 Message Date
  Megvii Engine Team f386381080 fix(imperative): fix inplace operation of optim 3 years ago
  Megvii Engine Team 67cfce9f2f fix(imperative/amp): add is_scalar check in elemwise and concat 3 years ago
  Megvii Engine Team d313f92610 fix(imperative/amp): fix format transformation for symbol trans 3 years ago
  Megvii Engine Team 261a5bce23 feat(imperative/amp): add dimshuffle in set_format for nhwc 3 years ago
  Megvii Engine Team cd26376549 style(imperative/amp): reformat code 3 years ago
  Megvii Engine Team fc633ce4ff fix(imperative/amp): fix custom grad in Subgraph 3 years ago
  Megvii Engine Team fc0f454685 fix(dnn/check_non_finite): adjust some details of CheckNonFinite 3 years ago
  Megvii Engine Team e393d1cf65 feat(mge/amp): add convert_format module for NHWC training 3 years ago
  Megvii Engine Team 533fb5bf49 feat(imperative): support formatted tensor and add special op rules 3 years ago
  Megvii Engine Team c2e9860feb chore(license): remove all license in file header 3 years ago
  Megvii Engine Team 3c3fc6f33c refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++ 3 years ago
  Megvii Engine Team 11d75fecb5 feat(dnn/check_non_finite): add batch check_non_finite 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