• jjsjann123's avatar
    [SyncBatchNorm] (#206) · 0a991543
    jjsjann123 authored
    supporting 2 dimensional input, resolving issue #194
    
    Implementation:
      for 2d input, switching channel_last flag to true for better memory access
    pattern in the kernel.
    0a991543
optimized_sync_batchnorm.py 4.15 KB