Allow warmup of models for when XLA or TRT optimization are enabled.
Allow warmup of models for when XLA or TRT optimization are enabled.