Home
last modified time | relevance | path

Searched defs:cur_stream (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/c10/xpu/test/impl/
H A DXPUStreamTest.cpp62 c10::xpu::XPUStream cur_stream = c10::xpu::getCurrentXPUStream(); in TEST() local
104 c10::xpu::XPUStream cur_stream = c10::xpu::getCurrentXPUStream(); in TEST() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcuda_stream_test.cpp109 at::cuda::CUDAStream cur_stream = at::cuda::getCurrentCUDAStream(); in TEST() local
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_ctx_manager.py436 def fn(x, cur_stream, new_stream): argument