Normalize an Expr to complete its shape and type. Parameters ---------- expr : Expr The input expr. Returns ------- ret : Expr The expr with normalized shape and type.
(self, expr: Expr)
| 660 | return self.add_func(func, self._func._name) |
| 661 | |
| 662 | def normalize(self, expr: Expr) -> Expr: |
| 663 | """Normalize an Expr to complete its shape and type. |
| 664 | |
| 665 | Parameters |
| 666 | ---------- |
| 667 | expr : Expr |
| 668 | The input expr. |
| 669 | |
| 670 | Returns |
| 671 | ------- |
| 672 | ret : Expr |
| 673 | The expr with normalized shape and type. |
| 674 | """ |
| 675 | return _ffi_api.BlockBuilderNormalize(self, expr) # type: ignore |
| 676 | |
| 677 | def get(self) -> tvm.IRModule: |
| 678 | """Return intermediate IRModule. For the situation where the IRModule is needed in the |
no outgoing calls