Skip to content

torch_to_nnef.nnef_graph

Core parsing and NNEF transformation module.

Classes:

Name Description
TorchToNGraphExtractor

Extract PyTorch Graph and build associated nnef_tools.model.Graph.

TorchToNGraphExtractor 

TorchToNGraphExtractor(model: Module, args: Tuple[Tensor, ...], inference_target: InferenceTarget, nnef_variable_naming_scheme: VariableNamingScheme = DEFAULT_VARNAME_SCHEME, forced_inputs_names: Optional[List[str]] = None, forced_outputs_names: Optional[List[str]] = None, check_io_names_qte_match: bool = True)

Extract PyTorch Graph and build associated nnef_tools.model.Graph.