| 113 | } |
| 114 | |
| 115 | StatusOr<std::unique_ptr<HloModule>> LoadModuleFromFile( |
| 116 | const string& path, hlo_module_loader_details::Config ovr_config, |
| 117 | string format, |
| 118 | const std::function<void(HloModuleConfig*)>& config_modifier_hook) { |
| 119 | string data; |
| 120 | if (format.empty()) { |
| 121 | format = string(tensorflow::io::Extension(path)); |
| 122 | } |
| 123 | TF_RETURN_IF_ERROR( |
| 124 | tensorflow::ReadFileToString(tensorflow::Env::Default(), path, &data)); |
| 125 | return LoadModuleFromData(data, format, ovr_config, config_modifier_hook); |
| 126 | } |
| 127 | |
| 128 | } // namespace xla |
no test coverage detected