elasticai.creator.nn.fixed_point.conv1d.layer.batch_normed_conv1d#

Module Contents#

Classes#

BatchNormedConv1d

API#

class elasticai.creator.nn.fixed_point.conv1d.layer.batch_normed_conv1d.BatchNormedConv1d(total_bits: int, frac_bits: int, in_channels: int, out_channels: int, signal_length: int, kernel_size: int | tuple[int], bn_eps: float = 1e-05, bn_momentum: float = 0.1, bn_affine: bool = True, stride: int | tuple[int] = 1, padding: int | tuple[int] = 0, bias: bool = True, device: Any = None)[source]#

Bases: elasticai.creator.nn.design_creator_module.DesignCreatorModule

property conv_weight: torch.Tensor#
property conv_bias: torch.Tensor | None#
property bn_weight: torch.Tensor#
property bn_bias: torch.Tensor#
forward(x: torch.Tensor) → torch.Tensor[source]#
create_design(name: str) → elasticai.creator.nn.fixed_point.conv1d.design.Conv1dDesign[source]#