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