elasticai.creator.nn.float.math_operations#

Module Contents#

Classes#

API#

class elasticai.creator.nn.float.math_operations.MathOperations(mantissa_bits: int, exponent_bits: int)[source]#

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

property largest_positive_value: float#
property smallest_negative_value: float#
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]#