operator

Classes

AirOperator

Intermediate Representation of an operator (node) in the graph.

Attributes:

• id (OpId) –

  Unique node identifier (e.g. 'conv1').

• op_type (AirOpType) –

  The operator unique type.

• input_ids (list[TensorId]) –

  Input tensor IDs.

• output_ids (list[TensorId]) –

  Output tensor IDs.

• named_tensors (dict[str, TensorId]) –

  Local tensor names to IDs mapping.

• options (AirOperatorOptions) –

  Typed options/config for this op.

Attributes

tensor_ids property

tensor_ids: list[TensorId]

All tensor IDs (inputs + outputs + named tensors) associated with this operator.

fan_in property

fan_in: int

How many input tensors this operator consumes.

fan_out property

fan_out: int

How many output tensors this operator produces.

Functions

print_info

print_info(verbose: int = 0)

Debug print the template values for the operator.