Sync tensor-size comments with the code. (Comments referred to earlier sizes, not the current ones.)