Searched refs:aoti_torch_get_current_cuda_stream (Results 1 – 2 of 2) sorted by relevance
50 aoti_torch_get_current_cuda_stream(int32_t device_index, void** ret_stream) { in aoti_torch_get_current_cuda_stream() function
613 aoti_torch_get_current_cuda_stream(int32_t device_index, void** ret_stream);