_init_from_local_shards (reference) in projects: aosp_15_r20

Searched refs:_init_from_local_shards (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A D	__init__.py	`407 return ShardedTensor._init_from_local_shards(`
H A D	api.py	`716 def _init_from_local_shards( member in ShardedTensor`
/aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/
H A D	test_sharded_tensor.py	`387 st_compare = ShardedTensor._init_from_local_shards( 3135 ShardedTensor._init_from_local_shards(`