UnshardHandle (definition) in projects: aosp_15_r20

Searched defs:UnshardHandle (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D	fully_shard.py	`381 class UnshardHandle: class`
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D	fully_sharded_data_parallel.py	`2055 class UnshardHandle: class`