[plugin] torch ddp plugin supports sharded model checkpoint

[ver217]

Overview

Torch DDP plugin does not overwrite save_sharded_model() method. This method should be overwritten in ditributed scenario.

Goal

1. Overwrite save_sharded_model() method.
2. Update unit tests