code: def pool_to_HW(shape, data_frmt):
      
      if len(shape) != 4:
          return shape # Not NHWC|NCHW, return as is
      if data_frmt == 'NCHW':
          return [shape[2], shape[3]]
      return [shape[1], shape[2]]
docs: Convert from NHWC|NCHW => HW