Struct FloatPrecision 

pub struct FloatPrecision { /* private fields */ }

Typed config for the float_precision transform.

Changes the float precision of a model (e.g. F32 to F16).

Example

use tract_api::DatumType;
model.transform(FloatPrecision::new(DatumType::F32, DatumType::F16))?;

Implementations

impl FloatPrecision

pub fn new(from: DatumType, to: DatumType) -> Self

pub fn include(self, patterns: Vec<String>) -> Self

Set include patterns — only nodes matching at least one pattern are translated.

pub fn exclude(self, patterns: Vec<String>) -> Self

Set exclude patterns — matching nodes are excluded from translation.

Trait Implementations

impl Clone for FloatPrecision

fn clone(&self) -> FloatPrecision

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FloatPrecision

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

Formats the value using the given formatter.

impl From<FloatPrecision> for TransformSpec

fn from(config: FloatPrecision) -> Self

Converts to this type from the input type.

impl Serialize for FloatPrecision

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl TransformConfig for FloatPrecision

fn name(&self) -> &'static str

The transform registry name (e.g. "pulse", "float_precision").

fn to_transform_string(&self) -> String

Produce the string the transform registry expects.