For nnvm index graph, we can easily fetch input index, attrs, …, of each node. The way I can figure out for similar task in Relay is to use ir_pass.post_order_visit. I have two questions:
- I can create a dictionary to record index information for each nodes. How can I fetch attrs for CallNode?
- What is TupleGetItemNode for? In resnet50 expr, all TupleGetItemNodes just pick index 0. If a TupleGetItemNode is an input node of a CallNode, can I replace this TupleGetItemNode with the node selected by it?
Or is there an easier way to do this?