7 Commits (261a5bce23b066b7c05c188ff638fa2c31b066f5)

Author SHA1 Message Date
  Megvii Engine Team 261a5bce23 feat(imperative/amp): add dimshuffle in set_format for nhwc 3 years ago
  Megvii Engine Team c9e56f4987 feat(imperative/amp): add dimshuffle before creating nhwc tensor 3 years ago
  Megvii Engine Team cd26376549 style(imperative/amp): reformat code 3 years ago
  Megvii Engine Team c28a875fac fix(imperative/amp): adapt new transformation 3 years ago
  Megvii Engine Team fd41302cc1 feat(imperative/amp): add set_format 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 3 years ago