xla

68daf61f - format the output model input (#6869)

format the output model input (#6869) Co-authored-by: root <root@a2-highgpu-1g-group-0vsl.us-central1-a.c.cloud-ml-benchmarking.internal>