Struct AutoMixedPrecision 

pub struct AutoMixedPrecision {
    pub policy: PrecisionPolicy,
}

Pass that rewrites a graph according to a PrecisionPolicy.

For each node:

1. Look up the target precision based on op kind.
2. Update node.shape.dtype to that precision.
3. If any input has a different dtype, insert a Cast node before it.

After this pass, every node knows its compute precision via its shape dtype. Backends dispatch kernels per-node.

Fields

policy: PrecisionPolicy

Implementations

impl AutoMixedPrecision

pub fn new(policy: PrecisionPolicy) -> AutoMixedPrecision

Trait Implementations

impl Pass for AutoMixedPrecision

fn name(&self) -> &str

Human-readable name for logging.

fn run(&self, graph: Graph) -> Graph

Transform the graph. Returns a new graph (or the same if no changes).