tree a9321e522c9ad30e8e6d9f29d607f48627d3f438
parent 934b4b6a35a22ec203be4dac3a3bdf4e64497fff
author Smit Hinsu <hinsu@google.com> 1596231440 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1596233617 -0700

Restrict GetDimensionSize HLO op result type to 32 bit integer

XLA implementation has this limitation and always uses 32 bit result for this instruction. This will cause mismatch between the result type in MLIR and XLA at the time of export.

This should be resolved once we have a special dialect mapping directly to HLOInstructionProto. Another option until then could be to introduce a pass to legalize mhlo itself to match XLA semantics.

PiperOrigin-RevId: 324286936
Change-Id: Ice7893f9920bbbc96936b90c8063248b1627e3e9