Home
last modified time | relevance | path

Searched defs:getDefaultStream (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/test/cpp_extensions/
Dmtia_extension.cpp56 c10::Stream getDefaultStream(c10::Device d) const override { in getDefaultStream() function
181 c10::Stream getDefaultStream(c10::DeviceIndex device) const override { in getDefaultStream() function
/external/pytorch/aten/src/ATen/detail/
DMTIAHooksInterface.h85 virtual c10::Stream getDefaultStream(DeviceIndex device) const { in getDefaultStream() function
/external/pytorch/c10/core/impl/
DVirtualGuardImpl.h45 Stream getDefaultStream(Device d) const override { in getDefaultStream() function
DDeviceGuardImplInterface.h112 virtual Stream getDefaultStream(Device) const { in getDefaultStream() function
/external/pytorch/aten/src/ATen/mps/
DMPSGuardImpl.h76 Stream getDefaultStream(Device d) const override { in getDefaultStream() function
/external/pytorch/c10/cuda/impl/
DCUDAGuardImpl.h62 Stream getDefaultStream(Device d) const override { in getDefaultStream() function
/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
DOpenRegHooks.cpp95 c10::Stream getDefaultStream(c10::Device d) const override { in getDefaultStream() function
/external/pytorch/aten/src/ATen/hip/impl/
DHIPGuardImplMasqueradingAsCUDA.h88 Stream getDefaultStream(Device d) const override { in getDefaultStream() function