pytorch_pfn_extras.onnx.pfto_exporter.export

Functions

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.is_in_onnx_export()	Returns whether it is in the middle of ONNX 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.Any(...)	Special type indicating an unconstrained type.
pytorch_pfn_extras.onnx.pfto_exporter.export.OperatorExportTypes(...)	Members:
pytorch_pfn_extras.onnx.pfto_exporter.export.partial	partial(func, *args, **keywords) - new function with partial application of the given arguments and keywords.