neural_compressor.model.torch_model

Class for PyTorch model.

Classes

PyTorchBaseModel	Build PyTorch base model.
PyTorchModel	Build PyTorchModel object.
PyTorchFXModel	Build PyTorchFXModel object.
IPEXModel	Build IPEXModel object.

Module Contents

class neural_compressor.model.torch_model.PyTorchBaseModel(model, **kwargs)

Build PyTorch base model.

class neural_compressor.model.torch_model.PyTorchModel(model, **kwargs)

Build PyTorchModel object.

class neural_compressor.model.torch_model.PyTorchFXModel(model, **kwargs)

Build PyTorchFXModel object.

class neural_compressor.model.torch_model.IPEXModel(model, **kwargs)

Build IPEXModel object.