4 Commits (15c6da62186197c72bb1eea5a14bb5baf8052870)

Author SHA1 Message Date
  Megvii Engine Team c28a875fac fix(imperative/amp): adapt new transformation 4 years ago
  Megvii Engine Team fd41302cc1 feat(imperative/amp): add set_format 4 years ago
  Megvii Engine Team fc633ce4ff fix(imperative/amp): fix custom grad in Subgraph 4 years ago
  Megvii Engine Team e393d1cf65 feat(mge/amp): add convert_format module for NHWC training 4 years ago