Searched refs:wide_parameter (Results 1 – 1 of 1) sorted by relevance
73 HloInstruction* wide_parameter = wide_while_body->parameter_instruction(0); in WidenWhileBody() local75 wide_parameter, narrow_shape.tuple_shapes_size()); in WidenWhileBody()85 wide_shape.tuple_shapes(i), wide_parameter, i))); in WidenWhileBody()