| 210 | } |
| 211 | |
| 212 | StatusOr<XlaComputation> Client::LoadSnapshot(const HloSnapshot& module) { |
| 213 | TF_RET_CHECK(module.has_hlo() && module.hlo().has_hlo_module()); |
| 214 | return XlaComputation(module.hlo().hlo_module()); |
| 215 | } |
| 216 | |
| 217 | StatusOr<ExecutionHandle> Client::Compile( |
| 218 | const XlaComputation& computation, absl::Span<const Shape> argument_shapes, |