tree c326deeb98bd56c4c1283a947fd5f867c33dd931
parent e432d9aafb14ed9cc288ce78a0386292a908f759
author A. Unique TensorFlower <gardener@tensorflow.org> 1646078805 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1646079352 -0800

Some legacy tests have inconsistencies layout when compiling the infeed
op and transferring infeed data. Adding this api allows the tests to
get the layout right for transferring data.

PiperOrigin-RevId: 431494217