Searched defs:wide_shape (Results 1 – 2 of 2) sorted by relevance
35 HloComputation* narrow_condition, const Shape& wide_shape) { in WidenWhileCondition()67 WidenWhileBody(HloComputation* narrow_body, const Shape& wide_shape) { in WidenWhileBody()
42 const Shape& wide_shape) { in WidenComputation()