Batch Norm 2D

Contents

Batch Norm 2D#

The batch_norm_2d module implements the BatchNorm2d PyTorch layer with the assumption that affine=False.

Note that the batch_norm_2d component is an internal hardware component and is exposed to the rest of the MASE software stack through the unified norm component.

Overview#

The batch_norm_2d is a fully pipelined module which follows the dataflow streaming protocol.

img

Contents