_wait_for_computation_stream (reference) in projects: aosp_15_r20

Searched refs:_wait_for_computation_stream (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D	fully_sharded_data_parallel.py	`67 _wait_for_computation_stream, 2088 _wait_for_computation_stream(`
H A D	_runtime_utils.py	`578 _wait_for_computation_stream( 1530 def _wait_for_computation_stream( function`