`elasticai.creator.base_modules.math_operations_torch`#

Module Contents#

Classes#

TorchMathOperations

API#

class elasticai.creator.base_modules.math_operations_torch.TorchMathOperations[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]#

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

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