jwfromm opened a new pull request #4899: [Relay][Frontend][Keras] NHWC import support. URL: https://github.com/apache/incubator-tvm/pull/4899 Currently the Keras importer can only output models with NCHW layout. This PR adds a layout argument to the Keras importer that allows NHWC to be used instead. This is useful for platforms that may have better NHWC than NCHW performance such as the Raspberry Pi. A few notes: * Xception is not being tested in NHWC mode as I found that it was causing a strange cuda error when targeting GPU. On CPU, it works fine however so I suspect this is an issue with the nhwc separable convolutions in Topi rather than the importer. * I made a few changes to the test script to allow eager execution (and TF2.0) to work.
---------------------------------------------------------------- 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
