You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2020/01/10 01:35:15 UTC
[GitHub] [incubator-tvm] optima2005 commented on issue #4659: [CONV3D]
remove incorrect schedule for conv3d_ndhwc
optima2005 commented on issue #4659: [CONV3D] remove incorrect schedule for conv3d_ndhwc
URL: https://github.com/apache/incubator-tvm/pull/4659#issuecomment-572833998
@masahi Yes. it would report error. Currently at least in TF frontend, the NHWC for conv2d, NDHWC for conv3d are forced to transform to NCHW/NCDHW layout. Please see [this](https://github.com/apache/incubator-tvm/blob/d6a23cf50fc5b9ddccfac679927f04b3baa1fe97/python/tvm/relay/frontend/tensorflow.py#L2291), [this](https://github.com/apache/incubator-tvm/blob/d6a23cf50fc5b9ddccfac679927f04b3baa1fe97/python/tvm/relay/frontend/tensorflow.py#L298) and [this](https://github.com/apache/incubator-tvm/blob/d6a23cf50fc5b9ddccfac679927f04b3baa1fe97/python/tvm/relay/frontend/tensorflow.py#L430). I guess that might be under perormance consideration. But a more user friendly approach may be needed to notice the user TVM not supporting those layouts under cuda target.
----------------------------------------------------------------
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:
users@infra.apache.org
With regards,
Apache Git Services