| 119 | } |
| 120 | |
| 121 | XlaOp IsInf(XlaOp operand) { |
| 122 | auto& b = *operand.builder(); |
| 123 | return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { |
| 124 | TF_RETURN_IF_ERROR(EnsureOperandIsRealFp("IsInf", operand)); |
| 125 | return IsPosInf(Abs(operand)); |
| 126 | }); |
| 127 | } |
| 128 | |
| 129 | XlaOp IsNan(XlaOp operand) { |
| 130 | auto& b = *operand.builder(); |