When you see the tensors changed from 4D to 5D, the corresponding conv2d op has 
already been changed from NCHW to NCHWc; otherwise the type won't match. This 
is called "alter op layout". Specifically, the function you pointed returns the 
altered NCHWc op:
https://github.com/apache/incubator-tvm/blob/main/python/tvm/topi/x86/conv2d_alter_op.py#L114

Accordingly, your graph changed from `4D -> conv2d_NCHW` to `4D -> 
layout_tranform -> 5D -> conv2d_NCHWc`.





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/where-does-the-layout-transform-of-each-op-happen-during-alter-op-layout-pass/8380/2)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/99a1dcccdfb5c46494624a06f976af89df8c9552d7a4abda5d4a932576286c19).

Reply via email to