nir.ir.pooling

Contents

`nir.ir.pooling`#

Module Contents#

Classes#

`SumPool2d`	Sum pooling layer in 2d.
`AvgPool2d`	Average pooling layer in 2d.

class nir.ir.pooling.SumPool2d#

Bases: nir.ir.node.NIRNode

Sum pooling layer in 2d.

kernel_size: numpy.ndarray#

stride: numpy.ndarray#

padding: numpy.ndarray#

input_type: Optional[Dict[str, numpy.ndarray]]#

output_type: Optional[Dict[str, numpy.ndarray]]#

metadata: Dict[str, Any]#

__post_init__()#

class nir.ir.pooling.AvgPool2d#

Bases: nir.ir.node.NIRNode

Average pooling layer in 2d.

kernel_size: numpy.ndarray#

stride: numpy.ndarray#

padding: numpy.ndarray#

__post_init__()#