test_handler

Classes

TestHandler

TestHandler(model: ModelT, subgraph: SubGraphT, operators: list[AotOperator], config: ConvertArgs)

Functions

compute_values

compute_values(model_content: bytes) -> dict[str, str]

Computes the values needed for generating the model source code.

Parameters:

• model_content

  (bytes) –

  The content of the model in bytes.

Returns:

• dict ( dict[str, str] ) –

  A dictionary containing the computed values for code generation.

generate_source_code

generate_source_code(model_content: bytes, save_path: Path)

Generate the source code for the main inference model.

Parameters:

• save_path

  (Path) –

  Path to save the generated code.

Functions