(irnode)
| 462 | |
| 463 | |
| 464 | def get_irnode_shape(irnode): |
| 465 | if isinstance(irnode, (list, tuple, ir.OpResultList)): |
| 466 | assert len(irnode) == 1 |
| 467 | irnode = irnode[0] |
| 468 | assert isinstance(irnode, (ir.RankedTensorType, ir.BlockArgument, ir.OpResult)) |
| 469 | if not isinstance(irnode, ir.RankedTensorType): |
| 470 | irnode = ir.RankedTensorType(irnode.type) |
| 471 | return tuple(irnode.shape) |
| 472 | |
| 473 | |
| 474 | def get_irnode_dtype(irnode): |