Struct GraphCostModel 

pub struct GraphCostModel {
    pub node_costs: HashMap<usize, OperationCost>,
    pub total_cost: OperationCost,
    pub metadata: HashMap<String, String>,
}

Cost model for an entire graph.

Fields

node_costs: HashMap<usize, OperationCost>

Cost annotations per node (indexed by node index)

total_cost: OperationCost

Total estimated cost

metadata: HashMap<String, String>

Cost model metadata

Implementations

impl GraphCostModel

pub fn new() -> Self

Create a new empty cost model.

pub fn set_node_cost(&mut self, node_idx: usize, cost: OperationCost)

Add a cost annotation for a node.

pub fn get_node_cost(&self, node_idx: usize) -> Option<&OperationCost>

Get the cost annotation for a node.

pub fn compute_total_cost(&mut self, graph: &EinsumGraph)

Compute the total cost based on node costs and graph structure.

pub fn with_metadata( self, key: impl Into<String>, value: impl Into<String>, ) -> Self

Add metadata to the cost model.

pub fn summary(&self) -> CostSummary

Get a summary of the cost model.

Trait Implementations

impl Clone for GraphCostModel

fn clone(&self) -> GraphCostModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphCostModel

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

Formats the value using the given formatter.

impl Default for GraphCostModel

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for GraphCostModel

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for GraphCostModel

fn eq(&self, other: &GraphCostModel) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for GraphCostModel

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for GraphCostModel