Searched refs:wide_shape (Results 1 – 1 of 1) sorted by relevance
31 HloComputation* narrow_condition, const Shape& wide_shape) { in WidenWhileCondition() argument38 HloInstruction::CreateParameter(0, wide_shape, "wide_param")); in WidenWhileCondition()63 WidenWhileBody(HloComputation* narrow_body, const Shape& wide_shape) { in WidenWhileBody() argument69 HloInstruction::CreateParameter(0, wide_shape, "wide_param")); in WidenWhileBody()82 i < wide_shape.tuple_shapes_size(); i++) { in WidenWhileBody()85 wide_shape.tuple_shapes(i), wide_parameter, i))); in WidenWhileBody()