Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DDeviceGuardImplInterface.h203 virtual void synchronizeStream(const Stream& /*stream*/) const { in synchronizeStream() function
302 void synchronizeStream(const Stream& /*stream*/) const override { in synchronizeStream() function
H A DVirtualGuardImpl.h81 void synchronizeStream(const Stream& stream) const override { in synchronizeStream() function
/aosp_15_r20/external/pytorch/c10/xpu/impl/
H A DXPUGuardImpl.h149 void synchronizeStream(const Stream& stream) const override { in synchronizeStream() function
/aosp_15_r20/external/pytorch/c10/cuda/impl/
H A DCUDAGuardImpl.h204 void synchronizeStream(const Stream& stream) const override { in synchronizeStream() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegHooks.cpp198 virtual void synchronizeStream(const c10::Stream& stream) const { in synchronizeStream() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmtia_extension.cpp118 void synchronizeStream(const c10::Stream& stream) const override { in synchronizeStream() function
/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPGuardImplMasqueradingAsCUDA.h207 void synchronizeStream(const Stream& stream) const override { in synchronizeStream() function
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp648 void ProcessGroupNCCL::WorkNCCL::synchronizeStream() { in synchronizeStream() function in c10d::ProcessGroupNCCL::WorkNCCL