elasticai.creator.base_modules.math_operations#

Module Contents#

Classes#

Quantize

MatMul

Add

Mul

API#

class elasticai.creator.base_modules.math_operations.Quantize[source]#

Bases: typing.Protocol

abstractmethod quantize(a: torch.Tensor) → torch.Tensor[source]#
class elasticai.creator.base_modules.math_operations.MatMul[source]#

Bases: typing.Protocol

abstractmethod matmul(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#
class elasticai.creator.base_modules.math_operations.Add[source]#

Bases: typing.Protocol

abstractmethod add(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#
class elasticai.creator.base_modules.math_operations.Mul[source]#

Bases: typing.Protocol

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