polygraphy.backend.onnx.loader.SaveOnnx Class Reference

Inheritance diagram for polygraphy.backend.onnx.loader.SaveOnnx:

Collaboration diagram for polygraphy.backend.onnx.loader.SaveOnnx:

Public Member Functions
def	__init__ (self, model, path=None)
def	__call__ (self)

Public Attributes
	path

Private Attributes
	_model

Constructor & Destructor Documentation

__init__()

def polygraphy.backend.onnx.loader.SaveOnnx.__init__	(		self,
			model,
			path = None
	)

Functor that saves an ONNX model to the specified path.

Args:
    model (Callable() -> onnx.ModelProto): A loader that can supply an ONNX model.
    path (str): Path at which to write the ONNX model.

Member Function Documentation

__call__()

def polygraphy.backend.onnx.loader.SaveOnnx.__call__

(

self

)

Saves an ONNX model to the specified path.

Returns:
    onnx.ModelProto: The model, after saving it.

Member Data Documentation

_model

polygraphy.backend.onnx.loader.SaveOnnx._model

private

path

polygraphy.backend.onnx.loader.SaveOnnx.path

---

The documentation for this class was generated from the following file:

• backend/onnx/loader.py