| 53 | } |
| 54 | |
| 55 | void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { |
| 56 | entry_computation_ = entry_computation; |
| 57 | config_.SetDefaultComputationLayout( |
| 58 | entry_computation_->ComputeProgramShape()); |
| 59 | input_output_alias_config_ = HloInputOutputAliasConfig( |
| 60 | entry_computation_->root_instruction()->shape()); |
| 61 | } |
| 62 | |
| 63 | HloComputation* HloModule::AddComputationInternal( |
| 64 | std::unique_ptr<HloComputation> computation, bool is_entry, |
nothing calls this directly
no test coverage detected