Searched refs:narrow_condition (Results 1 – 1 of 1) sorted by relevance
31 HloComputation* narrow_condition, const Shape& wide_shape) { in WidenWhileCondition() argument33 narrow_condition->parameter_instruction(0)->shape(); in WidenWhileCondition()36 HloComputation::Builder builder(StrCat("wide.", narrow_condition->name())); in WidenWhileCondition()46 return narrow_condition->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileCondition()54 {truncated_parameter}, narrow_condition)); in WidenWhileCondition()