code: def apply_tp(model: Transformer) -> None:
      _apply_tp_Transformer(model)
      for block in model.layers:
          # Apply to MLP
          _apply_tp_ffn(block.feed_forward)
          _apply_tp_attn(block.attention)
  
file_path: gpt-fast/tp.py
imports: import os
  from typing import List, Optional
  import torch
  import torch.distributed
  from torch import nn
  from torch.distributed import _functional_collectives
  from model import Attention, FeedForward, Transformer
  from quantize import WeightOnlyInt4Linear
label_desc: Use image node_id 10 for calling a global function with example usage: apply_tp(model) without return types