Skip to content

torch_to_nnef.inference_target.khronos

Classes:

Name Description
KhronosNNEF

Khronos Specification compliant NNEF asset build.

KhronosNNEF 

KhronosNNEF(version: Union[SemanticVersion, str], check_io: bool = True)

Bases: InferenceTarget

Khronos Specification compliant NNEF asset build.

in case of check_io=True we perform evaluation against nnef_tool nnef to pytorch converter. And access original and reloaded pytorch model provide same outputs

Methods:

Name Description
post_export

Check io via the Torch interpreter of NNEF-Tools.

post_export 

post_export(model_info: UnfoldModelInfo, nnef_graph: Graph, exported_filepath: Path, debug_bundle_path: Optional[Path] = None)

Check io via the Torch interpreter of NNEF-Tools.