| 31 | namespace xla { |
| 32 | |
| 33 | Status HloModuleImporter::Import(const xla::HloModule& module) { |
| 34 | // TODO(hinsu): Only import the entry computation here once all HLO ops with |
| 35 | // reference to other computation are updated to have a region instead of a |
| 36 | // function attribute. |
| 37 | for (const auto& computation : module.computations()) { |
| 38 | auto result = HloFunctionImporter::ImportFunction( |
| 39 | module_, &builder_, &function_map_, computation); |
| 40 | TF_RETURN_IF_ERROR(result.status()); |
| 41 | } |
| 42 | |
| 43 | return Status::OK(); |
| 44 | } |
| 45 | |
| 46 | Status HloModuleImporter::Import(const xla::HloModuleProto& module_proto) { |
| 47 | xla::DebugOptions debug_options; |
no test coverage detected