Home
last modified time | relevance | path

Searched refs:allocateTensorForReceiving (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/distributed/rpc/
Dtensorpipe_utils.h35 virtual at::DataPtr allocateTensorForReceiving(
Dtensorpipe_cuda.cpp98 at::DataPtr allocateTensorForReceiving( in allocateTensorForReceiving() function in torch::distributed::rpc::__anon87e9cf040111::TensorpipeCudaConverter
Dtensorpipe_utils.cpp75 at::DataPtr allocateTensorForReceiving( in allocateTensorForReceiving() function in torch::distributed::rpc::__anon68f10f380111::TensorpipeCpuConverter
271 at::DataPtr dataPtr = converter->allocateTensorForReceiving( in tensorpipeAllocate()