[churchx]

There are three bits important to rotational kinetic energy.Mass, it's distance from rotating center, and rpms. For engine flywheel it's all three. For driveshaft, only first to some extent with little left of later two. For wheels mostly first two.
It's not location pre or post transmission that increase or decrease need of energy spent to spin up some part. In this case it's difference of rpms that by chance are different for engine or post gearbox tranny parts, that affects that. Though drifeshaft also lacks diameter in addition to lower rpms it's spun at, for lightening to be very effective. If driveshaft had been spun at 10x faster rpms due different gearing from gearbox / final drive, results would be different. Rpms (and diameter), not location.
Think all of them as flywheels of different shapes/masses/spinning speeds and think what impacts energy needed to spin up flywheels in general and which each of them have of those properties.