Searched refs:while_body_indvar (Results 1 – 1 of 1) sorted by relevance
253 auto* while_body_indvar = NonConstantOperand(while_body_indvar_update); in PatternMatchLoopTripCount() local255 m::AddAnyOrder(m::Op().Is(while_body_indvar), in PatternMatchLoopTripCount()367 auto* while_body_indvar = NonConstantOperand(while_body_indvar_update); in ComputeWhileLoopTripCount() local389 while_body_indvar_update, {{while_body_indvar, &indvar_iter_val}}); in ComputeWhileLoopTripCount()457 auto* while_body_indvar = while_body_root->operand(indvar_index); in ComputeWhileLoopTripCountUpperBound() local458 if (while_body_indvar->opcode() != HloOpcode::kConstant) { in ComputeWhileLoopTripCountUpperBound()460 << while_body_indvar->ToString(); in ComputeWhileLoopTripCountUpperBound()467 TF_CHECK_OK(fake_input.CopyFrom(while_body_indvar->literal(), in ComputeWhileLoopTripCountUpperBound()