Searched defs:prepareTensorForSending (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ | ||
| H A D | tensorpipe_cuda.cpp | 75 std::optional<std::vector<char>> prepareTensorForSending( in prepareTensorForSending() function in torch::distributed::rpc::__anonf38404840111::TensorpipeCudaConverter |
| H A D | tensorpipe_utils.cpp | 39 std::optional<std::vector<char>> prepareTensorForSending( in prepareTensorForSending() function in torch::distributed::rpc::__anon47d1f4b80111::TensorpipeCpuConverter |