Struct FusionOp 

pub struct FusionOp {
    pub id: usize,
    pub kind: OpKind,
    pub inputs: Vec<usize>,
    pub output_shape: Vec<usize>,
    pub dtype_bytes: usize,
}

A single operation in the fusion graph.

id equals the op’s insertion index in its FusionGraph. inputs lists the op ids of the producers whose output tensors this op consumes; an empty inputs list marks a source (leaf) op.

Fields

id: usize

Stable identifier (equal to the insertion index in the graph).

kind: OpKind

The kind of operation.

inputs: Vec<usize>

Op ids of the producers this op reads from.

output_shape: Vec<usize>

Shape of this op’s output tensor (empty == scalar).

dtype_bytes: usize

Size in bytes of a single output element.

Implementations

impl FusionOp

pub fn output_elements(&self) -> Result<u64, GpuOptimError>

Number of elements in the output tensor (product of the shape dims).

Returns an error if the product overflows a u64.

pub fn output_bytes(&self) -> Result<u64, GpuOptimError>

Size in bytes of the output tensor (product(shape) * dtype_bytes).

Returns an error if the computation overflows a u64.

Trait Implementations

impl Clone for FusionOp

fn clone(&self) -> FusionOp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FusionOp

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

Formats the value using the given formatter.