Searched refs:aoti_torch_cuda_guard_set_index (Results 1 – 3 of 3) sorted by relevance
36 aoti_torch_cuda_guard_set_index(guard_.get(), device_index)); in set_index()
23 AOTITorchError aoti_torch_cuda_guard_set_index( in aoti_torch_cuda_guard_set_index() function
598 aoti_torch_cuda_guard_set_index(CUDAGuardHandle guard, int32_t device_index);