pub fn infer_output_shapes(
op: &StdTensorOp,
input_shapes: &[&[DimExpr]],
) -> Vec<Vec<DimExpr>>Expand description
Infer output shapes for a single instruction.
Returns a vector of shapes (one per output slot). For single-output ops, the vector has length 1. For multi-output linalg ops the vector has one entry per output.