Add test transform for model test helper for extension.
Add test transform for model test helper for extension.

PiperOrigin-RevId: 468552995

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo