Searched refs:wide_while_body (Results 1 – 1 of 1) sorted by relevance
66 HloComputation* wide_while_body = [&]() { in WidenWhileBody() local73 HloInstruction* wide_parameter = wide_while_body->parameter_instruction(0); in WidenWhileBody()77 wide_while_body->AddInstruction(HloInstruction::CreateCall( in WidenWhileBody()84 wide_while_body->AddInstruction(HloInstruction::CreateGetTupleElement( in WidenWhileBody()88 wide_while_body->set_root_instruction( in WidenWhileBody()93 return {{wide_while_body, std::move(inlined_instructions_map)}}; in WidenWhileBody()