code: def past_shape(*, hparams, batch_size=None, sequence=None):
      return [
          batch_size,
          hparams.n_layer,
          2,
          hparams.n_heads,
          sequence,
          hparams.n_embd // hparams.n_heads,
      ]
  
file_path: finetune/finetune/base_models/gpt2/featurizer.py
imports: import functools
  import tensorflow
  from finetune.util.shapes import shape_list, lengths_from_eos_idx
  from finetune.optimizers.recompute_grads import recompute_grad
  from finetune.nn.activations import gelu
  from finetune.base_models.gpt.featurizer import norm, dropout, get_pos_values
label_desc: Use image node_id 9 for calling a global function with example usage: past_shape() and returns: list