MLP · 131 params · 9.33KFLOP · 9 ops → 4 fused · CPU/XLA
input f32(32, 4)

memory
total         █████████████████████████  3.39 KB
params        ███▉·····················  524 B
input         ····███▌·················  512 B
intermediate  ········██████████████▎··  2 KB
output        ·······················██  384 B

layer               FLOPs              share  ops  output
MLP                 9.33K  ██████████ 100.0%    9  f32(32, 3)
  layer_1 Linear    4.62K  █████·····  49.6%    3  f32(32, 16)
  layer_2 Linear    3.17K  ·····███▍·  34.0%    3  f32(32, 3)