Searched defs:cudaAvailable (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ | ||
| H A D | request_callback_impl.cpp | 273 bool RequestCallbackImpl::cudaAvailable() const { in cudaAvailable() function in torch::distributed::rpc::RequestCallbackImpl |
| H A D | request_callback_no_python.cpp | 569 bool RequestCallbackNoPython::cudaAvailable() const { in cudaAvailable() function in torch::distributed::rpc::RequestCallbackNoPython |