NeuralNetOperationsConverter

Converter for neural network operations.

This converter handles neural network specific operations including:

  • Convolutional operations (conv2d) with proper attribute mapping

  • Pooling operations (maxPool2d, avgPool2d) with correct StableHLO mapping

  • Batch normalization and layer normalization operations

Supports operations as specified in Requirements 2.4:

  • conv2d with proper attribute mapping for strides, padding, dilation

  • Different pooling types (max, average) with correct StableHLO mapping

  • Batch normalization and layer normalization operations

Constructors

Link copied to clipboard
constructor()

Properties

Link copied to clipboard
open override val supportedOperations: Set<String>

Set of operation names this converter supports

Functions

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

Convert a graph node to StableHLO operations