code: def SyncBatchNorm(*args, **kwargs):
      """In cpu environment nn.SyncBatchNorm does not have kernel so use nn.BatchNorm2D instead"""
      if paddle.get_device() == "cpu":
          return nn.BatchNorm2D(*args, **kwargs)
      else:
          return nn.SyncBatchNorm(*args, **kwargs)
  
imports: from typing import Union, List, Tuple
  import paddle
  import paddle.nn
  import paddle.nn.functional
  from paddle.nn.layer import activation
  from paddle.nn import Conv2D, AvgPool2D
label_desc: Use image node_id 1 for calling a global function with example usage: SyncBatchNorm() and returns: nn, nn