src/transformers/models/opt/modeling_flax_opt.py/FlaxOPTLearnedPositionalEmbedding/__call__
class FlaxOPTLearnedPositionalEmbedding:    def __call__(self, positions):
        """`input_ids_shape` is expected to be [bsz x seqlen]."""

        return super().__call__(positions + self.offset)