main function

void main()

Implementation

void main() {
  print('========================================');
  print('   LAYER SPEED BENCHMARK (CPU ONLY)     ');
  print('========================================\n');

  int iterations = 100;

  // 1. MaxPooling2D
  MaxPooling2DLayer maxPool = MaxPooling2DLayer(poolSize: 2, stride: 2);
  Tensor<Matrix> poolInput = generateMatrix(128, 128);
  runBenchmark<Matrix, Matrix>('MaxPooling2D (128x128)', maxPool, poolInput, iterations);

  // 2. ReLULayerMatrix
  ReLULayerMatrix reluMatrix = ReLULayerMatrix();
  Tensor<Matrix> reluInput = generateMatrix(128, 128);
  runBenchmark<Matrix, Matrix>('ReLU Matrix (128x128)', reluMatrix, reluInput, iterations);

  // 3. MultiLSTMLayer (Assuming 3 stacked LSTMs)
  // Input: Seq=60, features=32 -> Hidden=64
  MultiTierLSTMLayer multiLstm = MultiTierLSTMLayer(64, tierClockCycles: [6,3]);
  Tensor<Matrix> multiLstmInput = generateMatrix(60, 32);
  runBenchmark<Matrix, Vector>('MultiLSTM (Seq:60, 3 layers, 64 hidden)', multiLstm, multiLstmInput, iterations);

  print('========================================');
  print('         BENCHMARK COMPLETE             ');
  print('========================================');
}