Grain size-dependent energy partition in phase transition of NiTi shape memory alloys studied by molecular dynamics simulation