NodeDeviceConfigurationProto#

class onnx_light.onnx.NodeDeviceConfigurationProto(*args, **kwargs)#

Defines a multi-device configuration proto for NodeProto.

ByteSize(self) → int#

Returns the serialized size in bytes, following the protobuf API.

CopyFrom(self, arg: onnx_light.onnx_py._onnxpy.NodeDeviceConfigurationProto, /) → None#

Copies one instance into this one.

ParseFromFile(self, name: str, options: object | None = None, external_data_file: str = '') → None#

Parses a binary file to fill this instance.

ParseFromString(self, data: bytes, options: object | None = None) → None#
ParseFromString(self, data: str, options: object | None = None) → None

Overloaded function.

  1. ParseFromString(self, data: bytes, options: object | None = None) -> None

Parses a sequence of bytes to fill this instance.

  1. ParseFromString(self, data: str, options: object | None = None) -> None

Parses a string to fill this instance.

SerializeSize(self, options: object | None = None) → onnx_light.onnx_py._onnxpy.SerializeSizeResult#

Returns the size once serialized without serializing.

SerializeToFile(self, name: str, options: object | None = None, external_data_file: str = '') → None#

Serializes this instance into a file. If external_data_size is not empty, big weights are stored in this (depending on options.raw_data_threshold). When writing to two files, temporary external-data metadata is cleared so the in-memory model stays unchanged.

SerializeToString(self, options: object | None = None) → bytes#

Serializes this instance into a sequence of bytes.

property configuration_id#

This field MUST be present for this version of the IR., ID of the configuration. MUST match the name of a DeviceConfigurationProto.

has_configuration_id(self) → bool#

Tells if ‘configuration_id’ has a value

has_pipeline_stage(self) → bool#

Tells if ‘pipeline_stage’ has a value.

has_sharding_spec(self) → bool#

Tells if ‘sharding_spec’ has a value.

property pipeline_stage#

Pipeline stage of this node.

property sharding_spec#

Sharding spec for the node.