src/transformers/models/whisper/modeling_tf_whisper.py/TFWhisperPositionalEmbedding/call
class TFWhisperPositionalEmbedding:    def call(self, input_ids, past_key_values_length=0):
        past_key_values_length = tf.cast(past_key_values_length, tf.int32)
        gather_indices = tf.range(tf.shape(input_ids)[-1], delta=1) + past_key_values_length
        return tf.gather(self.weight, gather_indices)