svdf

Classes

SvdfOperator

SvdfOperator(op: AirOperator, model: AirModel, platform: SocPlatform, prefix: str = 'aot', attributes: dict[str, str] = {})

SVDF operator.

Functions

compute_scratch_a_size

compute_scratch_a_size() -> int

Bytes for input_ctx->buf (buffer_a): (batches * num_filters * sizeof(int32))

compute_scratch_b_size

compute_scratch_b_size() -> int

Bytes for input_ctx->buf (buffer_b): (batches * num_filters * sizeof(int32))

compute_weight_sum_size

compute_weight_sum_size() -> int

Bytes for ctx->buf (kernel_sum_data): num_filters * sizeof(int32) if MVE, else 0.

compute_values

compute_values() -> dict[str, str]

Compute the values for the operator.

emit

emit(save_path: Path)

Generate the source code for the operator.

Functions