Skip to content

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[] .