Add another case where we should fallback to a batch size of 1 for inference.
Add another case where we should fallback to a batch size of 1 for inference.

PiperOrigin-RevId: 379388190

tensorflow/model-analysis,tensorflow/model-analysis,tensorflow/model-analysis,tensorflow/model-analysis