[SPARK-38679][CORE] Expose the number partitions in a stage to TaskContext#35995

vkorukanti wants to merge 5 commits intoapache:masterapache/spark:masterfrom

vkorukanti:task_numPartitionsvkorukanti/spark:task_numPartitionsCopy head branch name to clipboard