13 Commits (c9e56f4987e51c2fafd9c63c73ce0962c5d038bf)

Author SHA1 Message Date
  Megvii Engine Team c9e56f4987 feat(imperative/amp): add dimshuffle before creating nhwc tensor 3 years ago
  Megvii Engine Team d57a071271 feat(imperative/amp): add fallback for op not supported for nhwc tensor 3 years ago
  Megvii Engine Team 38a9aa9faf feat(imperative/amp): add auto dimshuffle for elemwise and concat 3 years ago
  Megvii Engine Team cd26376549 style(imperative/amp): reformat code 3 years ago
  Megvii Engine Team 6f0b582064 chore(imperative/amp): adapt dev 3 years ago
  Megvii Engine Team ee984e8608 fix(imperative/amp): fix distributed backward callback for nhwc amp 3 years ago
  Megvii Engine Team 15c6da6218 feat(imperative/amp): add nhwc support for adaptive pooling 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 673b295d75 feat(imperative/amp): remove conv_format and bn param_dim configs 3 years ago
  Megvii Engine Team 7e9aa742e6 feat(imperative/amp): enable auto_convert_format by default 3 years ago
  Megvii Engine Team 533fb5bf49 feat(imperative): support formatted tensor and add special op rules 4 years ago