[XLA] Tighten up shape inference rules
kRoundNearestAfz should be treated like kCeil and kFloor. kClz is only reasonable on integral types, not floating point or predicate types. kAbs and kSign are only reasonable on signed types and complex. PiperOrigin-RevId: 209978375
Loading
Please sign in to comment