skainet-compile-hlo/sk.ainet.compile.hlo.converters/MathOperationsConverter

MathOperationsConverter

class MathOperationsConverter : StableHloOperationConverter(source)

Converter for comprehensive mathematical operations.

This converter extends BasicMathConverter with additional mathematical operations and enhanced support for element-wise operations with broadcasting, mixed-type arithmetic with automatic type promotion, and proper operand ordering.

Supports operations as specified in Requirements 2.1 and 3.3:

Basic arithmetic: add, subtract, multiply, divide
Element-wise operations with broadcasting
Mixed-type arithmetic with automatic type promotion
Proper operand ordering and type consistency

Constructors

MathOperationsConverter

constructor()

Properties

supportedOperations

open override val supportedOperations: Set<String>

Set of operation names this converter supports

Functions

convert

open override fun convert(node: GraphNode, operands: List<String>, context: ConversionContext): ConversionResult

Convert a graph node to StableHLO operations