Optimize leaf transforms materialization

Optimize leaf materialization in EvaluationContext Use register for DStream leaves and an empty foreachPartition for other leaves instead of the current count() which adds overhead.