Home
last modified time | relevance | path

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

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