Searched defs:allocateTensorForReceiving (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ | ||
| H A D | tensorpipe_cuda.cpp | 98 at::DataPtr allocateTensorForReceiving( in allocateTensorForReceiving() function in torch::distributed::rpc::__anonf38404840111::TensorpipeCudaConverter |
| H A D | tensorpipe_utils.cpp | 75 at::DataPtr allocateTensorForReceiving( in allocateTensorForReceiving() function in torch::distributed::rpc::__anon47d1f4b80111::TensorpipeCpuConverter |