Skip to content

torch_to_nnef.torch_graph.ir_helpers

Functions:

Name Description
aten_name_to_torch_fn

Get aten cpp torch operator raw python binding.

dynamic_tensor_list_parse

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

t2n_extra_name_to_torch_fn

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

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: 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.