Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DVirtualGuardImpl.h85 void recordDataPtrOnStream(const c10::DataPtr& data_ptr, const Stream& stream) in recordDataPtrOnStream() function
H A DDeviceGuardImplInterface.h220 virtual void recordDataPtrOnStream(const c10::DataPtr&, const Stream&) const { in recordDataPtrOnStream() function
/aosp_15_r20/external/pytorch/c10/xpu/impl/
H A DXPUGuardImpl.h166 void recordDataPtrOnStream(const c10::DataPtr& data_ptr, const Stream& stream) in recordDataPtrOnStream() function
/aosp_15_r20/external/pytorch/c10/cuda/impl/
H A DCUDAGuardImpl.h222 void recordDataPtrOnStream(const c10::DataPtr& data_ptr, const Stream& stream) in recordDataPtrOnStream() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegHooks.cpp217 void recordDataPtrOnStream(const c10::DataPtr& data_ptr, const c10::Stream& stream) in recordDataPtrOnStream() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmtia_extension.cpp122 void recordDataPtrOnStream( in recordDataPtrOnStream() function
/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPGuardImplMasqueradingAsCUDA.h219 void recordDataPtrOnStream( in recordDataPtrOnStream() function