module tensor

aix.tensor

Source: packages/aix/src/aix/tensor.py:1

Tensor surface helpers for Arx.

Adapt Arx surface tensor syntax to IRx Tensor nodes while keeping user-facing shape and indexing rules local to Arx.

Classes

• TensorBinding — Static tensor binding metadata.

Functions

• attach_binding — Attach static tensor metadata to one AST node.

• binding_from_type — Build one static tensor binding from one declared type.

• build_literal_from_literal — Build one tensor literal from one nested literal value.

• coerce_expression — Coerce one parsed expression into one declared tensor type.

• default_value — Build one default tensor literal for one declared type.

• infer_literal — Infer one tensor literal directly from one literal value.

• is_tensor_type — Return whether one type is an Arx tensor surface type.

• literal_values — Return one flattened scalar payload from a tensor literal.

• runtime_tensor_type — Build one runtime-shaped tensor surface type.

• tensor_shape — Return the declared tensor shape when available.

• tensor_type — Build one static-shape tensor surface type.