module tensor

arx.tensor

Source: packages/arx/src/arx/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.

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

• 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.