| 2291 | } |
| 2292 | |
| 2293 | Status LayoutAssignment::Init() { |
| 2294 | computation_layouts_.clear(); |
| 2295 | conditional_mismatch_.clear(); |
| 2296 | *entry_computation_layout_ = saved_entry_computation_layout_; |
| 2297 | return Status::OK(); |
| 2298 | } |
| 2299 | |
| 2300 | Status LayoutAssignment::ClearPreviousPassSideEffects(HloModule* module) { |
| 2301 | VLOG(5) << "Clearing previous side effects"; |