Struct ShapeInferenceContext 

pub struct ShapeInferenceContext { /* private fields */ }

Shape inference context

Implementations

impl ShapeInferenceContext

pub fn new() -> Self

pub fn set_tensor_shape(&mut self, tensor_idx: usize, shape: TensorShape)

pub fn get_tensor_shape(&self, tensor_idx: usize) -> Option<&TensorShape>

pub fn infer_graph_shapes( &mut self, graph: &EinsumGraph, input_shapes: &HashMap<usize, TensorShape>, ) -> Result<(), String>

Infer shapes for all tensors in a graph

Trait Implementations

impl Default for ShapeInferenceContext

fn default() -> Self

Returns the “default value” for a type.