Allow to parameterize to which backend models will be exported {cpu,tpu}.
Allow to parameterize to which backend models will be exported {cpu,tpu}.

PiperOrigin-RevId: 254243467

google-research/tensor2robot