code: def make_temporal_pool(net, n_segment):
      if isinstance(net, torchvision.models.ResNet):
          net.layer2 = TemporalPool(net.layer2, n_segment)
      else:
          raise NotImplementedError
  
file_path: towhee/towhee/models/tsm/temporal_shift.py
imports: import torch
  from torch import nn
  import torch.nn.functional
  import torchvision
label_desc: Use image node_id 2 for calling a global function with example usage: make_temporal_pool(net, n_segment) without return types