Searched defs:wide_shape (Results 1 – 1 of 1) sorted by relevance
31 HloComputation* narrow_condition, const Shape& wide_shape) { in WidenWhileCondition()63 WidenWhileBody(HloComputation* narrow_body, const Shape& wide_shape) { in WidenWhileBody()