Searched refs:while_body_indvar (Results 1 – 1 of 1) sorted by relevance
395 auto* while_body_indvar = NonConstantOperand(while_body_indvar_update); in PatternMatchLoopTripCount() local397 m::AddAnyOrder(m::Op().Is(while_body_indvar), in PatternMatchLoopTripCount()509 auto* while_body_indvar = NonConstantOperand(while_body_indvar_update); in ComputeWhileLoopTripCount() local531 while_body_indvar_update, {{while_body_indvar, &indvar_iter_val}}); in ComputeWhileLoopTripCount()599 auto* while_body_indvar = while_body_root->operand(indvar_index); in ComputeWhileLoopTripCountUpperBound() local600 if (while_body_indvar->opcode() != HloOpcode::kConstant) { in ComputeWhileLoopTripCountUpperBound()602 << while_body_indvar->ToString(); in ComputeWhileLoopTripCountUpperBound()609 TF_CHECK_OK(fake_input.CopyFrom(while_body_indvar->literal(), in ComputeWhileLoopTripCountUpperBound()