Since we don't define layout attribute for such operators, nor the layout attribute in the tensor, there's no symbolic representation for their tensor layouts. In other words, you cannot get something like
``` %1 = nn.dense(...); /* output layout = NC */ ``` The way I can think of is just infer the layout manually from input tensor or the op with the layout attribute. Maybe this is also another opportunity to revisit the topic about whether to put layout attribute to the tensor, so that we could have a pass to calculate the input/output layouts of each op in symbolic way, and can further improve the layout conversion mechanism. cc @yzhliu @anijain2305 @tqchen --- [Visit Topic](https://discuss.tvm.apache.org/t/relay-topi-nn-ways-to-get-layout-of-each-operator/9310/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/9b1b0a45a2e18ed094f0a69a1d8e8f771af1f6a35935a3c9f7bbbd0ba3f3da27).
