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

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

Bases: typing.Protocol

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

Bases: typing.Protocol

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

Bases: typing.Protocol

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