composer.algorithms.factorize#

Modules

 composer.algorithms.factorize.factorize composer.algorithms.factorize.factorize composer.algorithms.factorize.factorize_core composer.algorithms.factorize.factorize_core composer.algorithms.factorize.factorize_modules composer.algorithms.factorize.factorize_modules

Decomposes linear operators into pairs of smaller linear operators.

See Factorize or the Method Card for details.

Functions

 apply_factorization Replaces Linear and Conv2d modules and with FactorizedLinear and FactorizedConv2d modules. factorize_conv2d Approximates a KxK convolution by factorizing it into a KxK convolution with fewer channels followed by a 1x1 convolution. factorize_matrix Approximates a matrix by factorizing it into a product of two smaller matrices. factorizing_could_speedup Whether factorizing a module a given amount could possibly yield a benefit.

Classes

 Factorize Decomposes linear operators into pairs of smaller linear operators. FactorizedConv2d Factorized replacement for torch.nn.Conv2d. FactorizedLinear Factorized replacement for torch.nn.Linear. LowRankSolution Bundles tensors used by a factorized linear operator.