TensorExpressionOps

Trait TensorExpressionOps 

Source
pub trait TensorExpressionOps<T: TensorElement> {
    // Required methods
    fn build_expression_graph(&self) -> ExpressionGraph;
    fn optimize_expressions(
        &self,
        config: OptimizerConfig,
    ) -> Result<OptimizationStats>;
}
Expand description

Extension trait to add expression optimization to tensors

Required Methods§

Source

fn build_expression_graph(&self) -> ExpressionGraph

Build an expression graph from tensor operations

Source

fn optimize_expressions( &self, config: OptimizerConfig, ) -> Result<OptimizationStats>

Optimize tensor expressions using the expression optimizer

Implementors§