ir_helpers

torch_to_nnef.torch_graph.ir_helpers

aten_name_to_torch_fn

aten_name_to_torch_fn(aten_name: str)

Get aten cpp torch operator raw python binding.

dynamic_tensor_list_parse

dynamic_tensor_list_parse(node_c_value: torch._C.Value)

Hold outputs of aten::chunk and other pytorch graph Tensor[] .

t2n_extra_name_to_torch_fn

t2n_extra_name_to_torch_fn(kind: str)

Resolve a t2n_extra::<name> op kind to its torch.ops entry.