Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/
H A D_common.py74 return ShardedTensor._init_from_local_tensor(
/aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/
H A Dtest_sharded_tensor.py2301 st = ShardedTensor._init_from_local_tensor(
2375 ShardedTensor._init_from_local_tensor(
2384 ShardedTensor._init_from_local_tensor(
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A Dapi.py776 def _init_from_local_tensor( member in ShardedTensor