pytorch_pfn_extras.onnx.pfto_exporter.export#

Functions

pytorch_pfn_extras.onnx.pfto_exporter.export.ONNXValueID(x)

pytorch_pfn_extras.onnx.pfto_exporter.export.TorchValueID(x)

pytorch_pfn_extras.onnx.pfto_exporter.export.cast(...)

Cast a value to a type.

pytorch_pfn_extras.onnx.pfto_exporter.export.contextmanager(func)

@contextmanager decorator.

pytorch_pfn_extras.onnx.pfto_exporter.export.export(...)

pytorch_pfn_extras.onnx.pfto_exporter.export.onnx_node_doc_string(...)

pytorch_pfn_extras.onnx.pfto_exporter.export.record(tag)

pytorch_pfn_extras.onnx.pfto_exporter.export.run_jit_pass(p, ...)

Classes

pytorch_pfn_extras.onnx.pfto_exporter.export.GraphContext(...)

Extra context for symbolic functions with all methods from torch.Graph.

pytorch_pfn_extras.onnx.pfto_exporter.export.OperatorExportTypes(...)

Members: