Skip to content

torch_to_nnef.op.aten

torch_to_nnef.op.aten

PyTorch Aten::* operators translation.

aten_to_nnef_tensor_and_ops

aten_to_nnef_tensor_and_ops(g, node, name_to_tensor, null_ref, torch_graph, inference_target: InferenceTarget) -> T.Optional[T.List[str]]

Main primitive dispatcher.

Allow to write in graph any not Quantized Operation from pytorch defined in node attribute.