| 415 | } |
| 416 | |
| 417 | StatusOr<std::vector<HloSnapshot>> ParseInputFile(const string& filename, |
| 418 | const Options& opts) { |
| 419 | std::vector<HloSnapshot> snapshots; |
| 420 | absl::string_view filename_view = filename; |
| 421 | if (absl::ConsumePrefix(&filename_view, "recordio_hlo_proto:")) { |
| 422 | return ParseRecordIoFile(filename_view, opts); |
| 423 | } |
| 424 | TF_ASSIGN_OR_RETURN(auto snapshot, ParseSingleHloFile(filename, opts)); |
| 425 | return std::vector<HloSnapshot>{std::move(snapshot)}; |
| 426 | } |
| 427 | |
| 428 | int RealMain(absl::Span<char* const> args, const Options& opts) { |
| 429 | LocalClient* client = ClientLibrary::LocalClientOrDie(); |
no test coverage detected