xla

aae5e4c2 - Use the equivalent Tensor shape for the input tensor. Added hack comment.