examples/research_projects/deebert/src/modeling_highway_bert.py/DeeBertModel/init_highway_pooler
class DeeBertModel:    def init_highway_pooler(self):
        self.encoder.init_highway_pooler(self.pooler)