(self)
| 549 | |
| 550 | # Return an MHLO IR of computation |
| 551 | def mhlo(self) -> ir.Module: |
| 552 | module_str = xla_extension.mlir.stablehlo_to_mhlo( |
| 553 | ir_utils.module_to_bytecode(self.stablehlo()) |
| 554 | ) |
| 555 | with self.stablehlo().context: |
| 556 | return ir.Module.parse(module_str) |
| 557 | |
| 558 | # Return a StableHLO IR of computation |
| 559 | def stablehlo(self) -> ir.Module: |