Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/_composable/
H A Dfully_shard.py18 _init_prefetching_state,
109 state = _init_prefetching_state(
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A Dfully_sharded_data_parallel.py50 _init_prefetching_state,
505 _init_prefetching_state(self, backward_prefetch, forward_prefetch)
H A D_init_utils.py512 def _init_prefetching_state( function