Fixed computation count mismatch between runtime and compiled layers.