Beya2019 opened a new pull request #4474: onnx frontend support layout choice depend on hardware target supported layout with NCHW and NHWC URL: https://github.com/apache/incubator-tvm/pull/4474 This submission adds a feature that onnx frontend support layout choice with NCHW and NHWC depended on hardware target supported layout . Now every frontend framework does not support layout choice depend. Such as, relay.frontend.from_onnx only support NCHW and relay.frontend.from_tensorflow only support NHWC. Given that TVM is supported NCHW and NHWC, and if our target hardware support NHWC， we should add NHWC layout in onnx frontend to produce the NHWC graph and finally compute on the target hardware. @vinx13 would you be able to take a look?
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
