Transformer Nits: reshape -> set_shape
Transformer Nits: reshape -> set_shape

PiperOrigin-RevId: 281872406

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