`elasticai.creator.nn.fixed_point.math_operations`#

Module Contents#

Classes#

MathOperations

API#

class elasticai.creator.nn.fixed_point.math_operations.MathOperations(config: elasticai.creator.nn.fixed_point.two_complement_fixed_point_config.FixedPointConfig)[source]#

Bases: elasticai.creator.base_modules.linear.MathOperations, elasticai.creator.base_modules.conv1d.MathOperations, elasticai.creator.base_modules.lstm_cell.MathOperations

quantize(a: torch.Tensor) → torch.Tensor[source]#

add(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#

matmul(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#

mul(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#