Skip to main content

Module shape_inference

Module shape_inference 

Source
Expand description

Shape inference for graph ops.

Given an OpKind and input shapes, computes the output shape. This is used by the graph builder to set TensorMeta on newly created nodes.

Enums§

ShapeError
Error returned when shapes are incompatible for an op.

Functions§

infer_shape
Infer the output shape for a given op and input shapes.