nnef_graph

torch_to_nnef.nnef_graph

Core parsing and NNEF transformation module.

TorchToNGraphExtractor

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

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