6 Commits (6f7649e9359ec8e1ac31cf3c7b45d7d9e0880737)

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 fc633ce4ff fix(imperative/amp): fix custom grad in Subgraph 3 years ago
  Megvii Engine Team e393d1cf65 feat(mge/amp): add convert_format module for NHWC training 4 years ago