Fix BERT SQUAD eager tests
Fix BERT SQUAD eager tests

PiperOrigin-RevId: 292947055

tombstone/models,alexgorban/models,alexgorban/models,alexgorban/models,tombstone/models,alexgorban/models,tombstone/models,tombstone/models,alexgorban/models,tombstone/models,tombstone/models