pytorch_pfn_extras.onnx#

Functions

pytorch_pfn_extras.onnx.annotate(**attrs)

Annotation parameters to the target function.

pytorch_pfn_extras.onnx.apply_annotation(fn, ...)

Annotation applier to the target function

pytorch_pfn_extras.onnx.as_output(name, value)

pytorch_pfn_extras.onnx.export(model, args, f)

Export model into ONNX Graph.

pytorch_pfn_extras.onnx.export_testcase(...)

Export model and I/O tensors of the model in protobuf format.

pytorch_pfn_extras.onnx.grad(output, inputs)

pytorch_pfn_extras.onnx.is_large_tensor(...)

pytorch_pfn_extras.onnx.load_model(f[, ...])

Load model from ONNX file.

pytorch_pfn_extras.onnx.no_grad(fn, *args, ...)

pytorch_pfn_extras.onnx.scoped_anchor(**attrs)

Add anchor node to the scoped modules

Modules

pytorch_pfn_extras.onnx.annotate(**attrs)

Annotation parameters to the target function.

pytorch_pfn_extras.onnx.export_testcase(...)

Export model and I/O tensors of the model in protobuf format.

pytorch_pfn_extras.onnx.load

pytorch_pfn_extras.onnx.pfto_exporter

pytorch_pfn_extras.onnx.strip_large_tensor

pytorch_pfn_extras.onnx.symbolic_registry

pytorch_pfn_extras.onnx.unstrip_tensor