Home
last modified time | relevance | path

Searched defs:deviceSynchronize (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/
H A DXPUHooks.cpp64 void XPUHooks::deviceSynchronize(DeviceIndex device_index) const { in deviceSynchronize() function in at::xpu::detail::XPUHooks
/aosp_15_r20/external/pytorch/aten/src/ATen/detail/
H A DXPUHooksInterface.h57 virtual void deviceSynchronize(DeviceIndex /*device_index*/) const { in deviceSynchronize() function
H A DMTIAHooksInterface.h49 virtual void deviceSynchronize(c10::DeviceIndex device_index) const { in deviceSynchronize() function
H A DMPSHooksInterface.h40 virtual void deviceSynchronize() const { in deviceSynchronize() function
H A DCUDAHooksInterface.h195 virtual void deviceSynchronize(DeviceIndex /*device_index*/) const { in deviceSynchronize() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmtia_extension.cpp153 void deviceSynchronize(c10::DeviceIndex device_index) const override { in deviceSynchronize() function
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DCUDAHooks.cpp457 void CUDAHooks::deviceSynchronize(DeviceIndex device_index) const { in deviceSynchronize() function in at::cuda::detail::CUDAHooks