Enable streaming support for pos_embedding and norm in SubSampleConvProjection. Also create streaming layers in setup() instead of override (it will allow to to control supported layer types with multiple options for streaming).
Enable streaming support for pos_embedding and norm in SubSampleConvProjection. Also create streaming layers in setup() instead of override (it will allow to to control supported layer types with multiple options for streaming).

PiperOrigin-RevId: 478058259

google/praxis,google/praxis