tree 5107cc7f40ab059577a84909490de5cb4f9ba630
parent e0cb70344e46276b37d65824f95eca478080de4a
author Anish Tondwalkar <atondwal@google.com> 1646415628 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1646415827 -0800

Use correct dimensions when legalizing tf.IRFFT to mhlo

This fixes a bug in tf legalization: While we need to calculate the sizes of tensor slices using after running the IRFFT, the fft_length before running the non-inverse FFT is the argument to mhlo.fft, according to the HLO specification.

This fix unblocks merging the mhlo.fft verifier.

PiperOrigin-RevId: 432466930