code: def kaiming_uniform_(
      tensor, a=0, mode="fan_in", nonlinearity="leaky_relu"
  ):
      weight_tensor_nchw = tensor
      nn.init.kaiming_uniform_(
          weight_tensor_nchw, a=a, mode=mode, nonlinearity=nonlinearity
      )
  
imports: import functools
  import torch
  import torch.distributed
  from torch import nn
  from apex import check_cudnn_version_and_warn
  import fast_bottleneck
  import nccl_p2p_cuda
label_desc: Use image node_id 1 for calling a global function with example usage: kaiming_uniform_(tensor, a, mode, nonlinearity) without return types