For TF models only.