Searched refs:body_elem (Results 1 – 1 of 1) sorted by relevance
445 auto* body_elem = while_body_root->operand(i); in TryRemoveRepeatedWhileTupleIndices() local446 if (body_elem->opcode() == HloOpcode::kGetTupleElement && in TryRemoveRepeatedWhileTupleIndices()447 body_elem->operand(0) == while_body->parameter_instruction(0)) { in TryRemoveRepeatedWhileTupleIndices()448 if (body_elem->tuple_index() != i) { in TryRemoveRepeatedWhileTupleIndices()450 << body_elem->tuple_index() << " i " << i; in TryRemoveRepeatedWhileTupleIndices()460 VLOG(2) << "body_elem " << body_elem->ToString() << " pivot_body_elem " in TryRemoveRepeatedWhileTupleIndices()512 auto* body_elem = while_body_root->operand(i); in TryRemoveConstantParams() local514 body_elem->opcode() == HloOpcode::kConstant && in TryRemoveConstantParams()515 init_elem->literal() == body_elem->literal()) { in TryRemoveConstantParams()