Hi, I have a question about the layout, does tvm has a method to get every relay::callnode and relay::function’s input, output, weights layout. Thanks!
Only certain ops are aware of layout information, so there is no layout information attached to relay::Function inputs and most of relay::Call node. For layout-aware ops, you can query layout information via op attributes. See example in https://github.com/apache/tvm/blob/main/python/tvm/relay/op/contrib/cutlass.py#L119-L120
Thanks for your reply, assuming that there is no layout transformation, I want to develop a
function to collect all CallNode and Function input output layout in a model, any advice ?