Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Jiarui Fang's avatar
    [zero] sharded tensor (#305) · 80364c76
    Jiarui Fang authored 
    * init shard param from shape tuple

* add more unitest for shard param

* add set_payload method for ShardedParam

* [zero] add shareded tensor class

* polish code
    80364c76
test_shard_param.py 3.16 KB