fhelium.engine.ntt.backends.indexed_radix2
Indexed radix-2 NTT execution.
IndexedRadix2NttBackend
class View source
python
IndexedRadix2NttBackend(*, policy: IndexedRadix2Policy, ntt_tables: IndexedRadix2Tables, rns_params: torch.Tensor)1
Execute the indexed radix-2 schedule on CPU or CUDA.
forward_montgomery_
method
python
def forward_montgomery_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
forward_to_montgomery_
method
python
def forward_to_montgomery_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
forward_to_montgomery
method
python
def forward_to_montgomery(operand: torch.Tensor, parameter_row_start: int) -> torch.Tensor: ...1
inverse_montgomery_
method
python
def inverse_montgomery_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
inverse_to_standard_lazy_
method
python
def inverse_to_standard_lazy_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
inverse_to_standard_
method
python
def inverse_to_standard_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1
inverse_to_centered_
method
python
def inverse_to_centered_(operand: torch.Tensor, parameter_row_start: int) -> None: ...1