MomentumPropagation implementation