Searched refs:narrow_body (Results 1 – 1 of 1) sorted by relevance
63 WidenWhileBody(HloComputation* narrow_body, const Shape& wide_shape) { in WidenWhileBody() argument64 const Shape& narrow_shape = narrow_body->parameter_instruction(0)->shape(); in WidenWhileBody()67 HloComputation::Builder builder(StrCat("wide.", narrow_body->name())); in WidenWhileBody()70 return narrow_body->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileBody()78 narrow_shape, {truncated_parameter}, narrow_body)); in WidenWhileBody()