tensor_handler

Classes

TensorHandler

TensorHandler(context: CodeGenContext)

Handles the generation of tensor-related source code.

Functions

compute_values

compute_values() -> dict[str, str]

Compute the values for the operator source code template.

Returns:

• dict[str, str] –

  dict[str, str]: Dictionary of values for the operator template.

emit

emit(save_path: Path)

Generate the source code for resource variables.

Parameters:

• save_path

  (Path) –

  Path to save the generated code.

Functions