Struct DeviceConfig 

pub struct DeviceConfig {
    pub float_dtype: Option<FloatDType>,
    pub int_dtype: Option<IntDType>,
    pub bool_dtype: Option<BoolDType>,
}

Configuration options used to initialize a device.

Unlike DeviceSettings, this type represents partial user-provided configuration and does not require all settings to be specified.

Any unspecified options will be resolved to device-specific defaults when the device is initialized.

Use Device::configure to apply this configuration to a device.

Fields

float_dtype: Option<FloatDType>

Default floating-point data type.

int_dtype: Option<IntDType>

Default integer data type.

bool_dtype: Option<BoolDType>

Default boolean data type.

Implementations

impl DeviceConfig

pub fn new( float_dtype: Option<FloatDType>, int_dtype: Option<IntDType>, bool_dtype: Option<BoolDType>, ) -> Self

Constructs a new DeviceConfig.

impl DeviceConfig

pub fn float_dtype(self, dtype: impl Into<FloatDType>) -> Self

Sets the default floating-point data type for tensors created on the device.

pub fn int_dtype(self, dtype: impl Into<IntDType>) -> Self

Sets the default integer data type for tensors created on the device.

pub fn bool_dtype(self, dtype: impl Into<BoolDType>) -> Self

Sets the default boolean data type storage precision for tensors created on the device.

Trait Implementations

impl Clone for DeviceConfig

fn clone(&self) -> DeviceConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DeviceConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for DeviceConfig

fn default() -> DeviceConfig

Returns the “default value” for a type.

impl From<(FloatDType, IntDType)> for DeviceConfig

fn from(value: (FloatDType, IntDType)) -> Self

Converts to this type from the input type.

impl From<BoolStore> for DeviceConfig

fn from(value: BoolDType) -> Self

Converts to this type from the input type.

impl From<FloatDType> for DeviceConfig

fn from(value: FloatDType) -> Self

Converts to this type from the input type.

impl From<IntDType> for DeviceConfig

fn from(value: IntDType) -> Self

Converts to this type from the input type.