Home
last modified time | relevance | path

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

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