Struct GraphDataflow 

pub struct GraphDataflow {
    pub live_tensors: HashMap<usize, HashSet<usize>>,
    pub dependencies: HashMap<usize, HashSet<usize>>,
    pub uses: HashMap<usize, HashSet<usize>>,
}

Dataflow analysis for einsum graphs.

Fields

live_tensors: HashMap<usize, HashSet<usize>>

Live tensors at each node

dependencies: HashMap<usize, HashSet<usize>>

Tensor dependencies

uses: HashMap<usize, HashSet<usize>>

Reverse dependencies (uses)

Implementations

impl GraphDataflow

pub fn new() -> Self

Create a new graph dataflow analysis.

pub fn is_tensor_live(&self, node: usize, tensor: usize) -> bool

Check if a tensor is live at a node.

pub fn get_dependencies(&self, tensor: usize) -> HashSet<usize>

Get dependencies of a tensor.

pub fn get_uses(&self, tensor: usize) -> HashSet<usize>

Get uses of a tensor.

Trait Implementations

impl Clone for GraphDataflow

fn clone(&self) -> GraphDataflow

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphDataflow

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

Formats the value using the given formatter.

impl Default for GraphDataflow

fn default() -> Self

Returns the “default value” for a type.