Home
last modified time | relevance | path

Searched defs:exchangeStream (Results 1 – 13 of 13) sorted by relevance

/external/pytorch/aten/src/ATen/native/metal/
DMetalGuardImpl.cpp35 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
/external/pytorch/aten/src/ATen/native/vulkan/
DVulkanGuardImpl.cpp40 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
/external/pytorch/c10/core/impl/
DFakeGuardImpl.h51 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
DVirtualGuardImpl.h52 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
DDeviceGuardImplInterface.h273 Stream exchangeStream(Stream) const noexcept override { in exchangeStream() function
/external/pytorch/aten/src/ATen/mps/
DMPSGuardImpl.h81 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
/external/pytorch/c10/xpu/impl/
DXPUGuardImpl.h61 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
/external/pytorch/torch/csrc/lazy/core/
Dtensor_impl.cpp54 c10::Stream exchangeStream(c10::Stream _unused) const noexcept override { in exchangeStream() function
/external/pytorch/c10/cuda/impl/
DCUDAGuardImpl.h73 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
/external/pytorch/test/cpp_extensions/
Dmaia_extension.cpp90 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function
Dmtia_extension.cpp65 c10::Stream exchangeStream(c10::Stream s) const noexcept override { in exchangeStream() function
/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
DOpenRegHooks.cpp123 c10::Stream exchangeStream(c10::Stream s) const noexcept override { in exchangeStream() function
/external/pytorch/aten/src/ATen/hip/impl/
DHIPGuardImplMasqueradingAsCUDA.h97 Stream exchangeStream(Stream s) const noexcept override { in exchangeStream() function