Home
last modified time | relevance | path

Searched defs:hasPrimaryContext (Results 1 – 11 of 11) sorted by relevance

/external/pytorch/aten/src/ATen/mps/
DMPSHooks.h54 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
/external/pytorch/aten/src/ATen/detail/
DPrivateUse1HooksInterface.h37 virtual bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
DHIPHooksInterface.h62 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
DXPUHooksInterface.h69 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
DMTIAHooksInterface.h57 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
DMPSHooksInterface.h94 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
DCUDAHooksInterface.h116 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
/external/pytorch/c10/cuda/
DCUDAFunctions.cpp176 bool (*hasPrimaryContext)(DeviceIndex) = dummyHasPrimaryContext; variable
184 bool hasPrimaryContext(DeviceIndex device_index) { in hasPrimaryContext() function
/external/pytorch/aten/src/ATen/xpu/detail/
DXPUHooks.cpp83 bool XPUHooks::hasPrimaryContext(DeviceIndex device_index) const { in hasPrimaryContext() function in at::xpu::detail::XPUHooks
/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
DOpenRegHooks.cpp21 bool hasPrimaryContext(c10::DeviceIndex device_index) const override { in hasPrimaryContext() function
/external/pytorch/aten/src/ATen/cuda/detail/
DCUDAHooks.cpp248 bool CUDAHooks::hasPrimaryContext(DeviceIndex device_index) const { in hasPrimaryContext() function in at::cuda::detail::CUDAHooks