Searched defs:CUDAInitMode (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_fsdp.py | 76 class CUDAInitMode(Enum): class
240 cuda_init_mode: CUDAInitMode,
306 cuda_init_mode: CUDAInitMode,
389 cuda_init_mode: CUDAInitMode,
435 cuda_init_mode: CUDAInitMode,
483 cuda_init_mode: CUDAInitMode,
517 cuda_init_mode: CUDAInitMode,
565 cuda_init_mode: CUDAInitMode,
683 cuda_init_mode: CUDAInitMode = CUDAInitMode.CUDA_AFTER,
715 cuda_init_mode: CUDAInitMode,
[all …]
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_use_orig_params.py | 118 cuda_init_mode: CUDAInitMode,
336 cuda_init_mode: CUDAInitMode,
|
| H A D | test_wrap.py | 318 cuda_init_mode: CUDAInitMode,
|