Home
last modified time | relevance | path

Searched defs:getStreamFromGlobalPool (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegHooks.cpp103 c10::Stream getStreamFromGlobalPool(c10::Device d, bool isHighPriority = false) const override { in getStreamFromGlobalPool() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmtia_extension.cpp59 c10::Stream getStreamFromGlobalPool( in getStreamFromGlobalPool() function