fhelium.backend.ntt.operations
Native implementations of NTT transitions.
NativeNttImplementation
class View source
python
NativeNttImplementation(name: str, required_backend_name: str | None, supports_in_place: bool = True, operation_types: tuple[type[Operation], ...] = _NTT_OPERATION_TYPES)1
Bases: _TensorOperandImplementation
Execute logical NTT operations through one named schedule executor.
Attributes
| Name | Type | Default/value |
|---|---|---|
name | str | |
required_backend_name | str | None | |
supports_in_place | bool | True |
operation_types | tuple[type[Operation], ...] | _NTT_OPERATION_TYPES |
select_ntt_schedule
method
python
def select_ntt_schedule(**facts): ...1
prepare_operation
method
python
def prepare_operation(operation: Operation): ...1
execute
method
python
def execute(invocation: OperationInvocation, values: tuple[torch.Tensor, ...], resources: tuple[BoundResource, ...], *, in_place: bool) -> tuple[torch.Tensor, ...]: ...1