Commit Graph

1 Commits

Author SHA1 Message Date
Benjamin Kramer 03d2cb606d [mhlo] Make sure reifyResultTypes returns a tensor of index
Dynamic broadcast/reshape/iota take i32/i64 shape inputs, but users of
reification expect index shapes. Insert an appropriate cast if necessary.

PiperOrigin-RevId: 380613128
2021-06-21 10:42:38 -07:00