| 95 | } |
| 96 | |
| 97 | StatusOr<std::shared_ptr<HloModule>> GetHloModule( |
| 98 | const XlaComputation& computation) { |
| 99 | TF_ASSIGN_OR_RETURN(const HloModuleConfig module_config, |
| 100 | HloModule::CreateModuleConfigFromProto( |
| 101 | computation.proto(), GetDebugOptionsFromFlags())); |
| 102 | TF_ASSIGN_OR_RETURN( |
| 103 | std::unique_ptr<HloModule> module, |
| 104 | HloModule::CreateFromProto(computation.proto(), module_config)); |
| 105 | return std::shared_ptr<HloModule>(std::move(module)); |
| 106 | } |
| 107 | |
| 108 | // Converts a computation to textual HLO form. |
| 109 | StatusOr<std::string> GetComputationHloText(const XlaComputation& computation) { |