neuromancer.slim.butterfly package

Submodules

• neuromancer.slim.butterfly.benchmark module
• neuromancer.slim.butterfly.butterfly module
  • Butterfly
    • Butterfly.extra_repr()
    • Butterfly.forward()
    • Butterfly.reset_parameters()
• neuromancer.slim.butterfly.butterfly_multiply module
  • ButterflyFactorMult
    • ButterflyFactorMult.backward()
    • ButterflyFactorMult.forward()
  • ButterflyMult
    • ButterflyMult.backward()
    • ButterflyMult.forward()
  • ButterflyMultInplace
    • ButterflyMultInplace.backward()
    • ButterflyMultInplace.forward()
  • ButterflyMultUntied
    • ButterflyMultUntied.backward()
    • ButterflyMultUntied.forward()
  • butterfly_mult()
  • butterfly_mult_factors()
  • butterfly_mult_torch()
  • butterfly_mult_untied()
  • butterfly_mult_untied_torch()
• neuromancer.slim.butterfly.complex_utils module
  • ComplexMatmulNp
    • ComplexMatmulNp.backward()
    • ComplexMatmulNp.forward()
  • ComplexMul
    • ComplexMul.backward()
    • ComplexMul.forward()
  • Conjugate
    • Conjugate.backward()
    • Conjugate.forward()
  • complex_matmul_torch()
  • complex_mul_numpy()
  • complex_mul_torch()
  • conjugate_torch()
  • cupy2torch()
  • real_to_complex()
  • test_complex_mm()
  • test_complex_mul()
  • torch2cupy()
  • torch2numpy()
• neuromancer.slim.butterfly.permutation module
  • FixedPermutation
    • FixedPermutation.forward()
  • Permutation
    • Permutation.argmax()
    • Permutation.extra_repr()
    • Permutation.forward()
  • PermutationFactor
    • PermutationFactor.argmax()
    • PermutationFactor.extra_repr()
    • PermutationFactor.forward()
• neuromancer.slim.butterfly.permutation_multiply module
  • PermutationFactorEvenOddMult
    • PermutationFactorEvenOddMult.backward()
    • PermutationFactorEvenOddMult.forward()
  • PermutationFactorReverseMult
    • PermutationFactorReverseMult.backward()
    • PermutationFactorReverseMult.forward()
  • permutation_mult()
  • permutation_mult_factors()
  • permutation_mult_single()
  • permutation_mult_single_factor()
  • permutation_mult_single_factor_torch()
  • permutation_mult_torch()
• neuromancer.slim.butterfly.utils module
  • bitreversal_permutation()

Module contents