Imag(Complex(r, i)) -> i
| 2661 | |
| 2662 | // Imag(Complex(r, i)) -> i |
| 2663 | Status AlgebraicSimplifierVisitor::HandleImag(HloInstruction* imag) { |
| 2664 | HloInstruction* op; |
| 2665 | if (Match(imag, m::Imag(m::Complex(m::Op(), m::Op(&op))))) { |
| 2666 | return ReplaceInstruction(imag, op); |
| 2667 | } |
| 2668 | return Status::OK(); |
| 2669 | } |
| 2670 | |
| 2671 | Status AlgebraicSimplifierVisitor::HandleIota(HloInstruction* instruction) { |
| 2672 | // iota -> zero if the iota dimension never produces an element other than |