Searched refs:init_from_local_shards (Results 1 – 5 of 5) sorted by relevance
2437 st = sharded_tensor.init_from_local_shards(2644 st = sharded_tensor.init_from_local_shards(2694 st = sharded_tensor.init_from_local_shards(2704 st = sharded_tensor.init_from_local_shards(2717 st = sharded_tensor.init_from_local_shards(2754 st = sharded_tensor.init_from_local_shards(2769 st = sharded_tensor.init_from_local_shards(2786 st = sharded_tensor.init_from_local_shards(2816 st = sharded_tensor.init_from_local_shards(2830 st = sharded_tensor.init_from_local_shards([all …]
358 def init_from_local_shards( function
24 init_from_local_shards,442 sharded_tensor = init_from_local_shards(
15 init_from_local_shards,1291 sharded_tensor = init_from_local_shards(
37 init_from_local_shards,1799 st = init_from_local_shards(local_shards, [10, 10])