afe.ir.transform_ir_generator

Generation of AwesomeNodes from a tensor transform.

Classes

TransformIRGenerationContext

Context for generating AwesomeNodes for a tensor transform.

TransformIRGenerator

Generator of AwesomeNodes for a tensor transform. Class instances

Module Contents

class afe.ir.transform_ir_generator.TransformIRGenerationContext[source]

Context for generating AwesomeNodes for a tensor transform.

class afe.ir.transform_ir_generator.TransformIRGenerator[source]

Generator of AwesomeNodes for a tensor transform. Class instances are produced from tensor transforms.

abstract generate(generation_info: TransformIRGenerationContext, input_type: afe.ir.tensor_type.TensorType, input_name: afe.ir.defines.NodeName | None, output_name: afe.ir.defines.NodeName | None) → Tuple[afe.ir.defines.NodeName, afe.ir.defines.NodeName, afe.ir.tensor_type.TensorType, List[afe.ir.node.AwesomeNode]][source]

Generate AwesomeNodes for the transform.

A caller must pass None for at least one of input_name and output_name.

Parameters:

  • generation_info – The context where generation occurs

  • input_type – Type of the input node.

  • input_name – Name of the input node. If not given, and a name is needed, a name will be created.

  • output_name – Name of the output node. If not given, and a name is needed, a name will be created.

Returns:

Tuple of input node name, output node name, output type, and newly created nodes. The nodes should be inserted into an AwesomeNet.