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

Module Contents#

Classes#

MathOperations

API#

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

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

round(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]#

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