Update weight actions, add partialSum param.