[SPARK-4092] Fix InputMetrics for coalesce

Fixes the input metrics for using coalesce.