Searched refs:cloneSparseTensors (Results 1 – 4 of 4) sorted by relevance
58 auto v1 = torch::distributed::rpc::cloneSparseTensors({big}); in TEST()62 auto v2 = torch::distributed::rpc::cloneSparseTensors({tiny}); in TEST()67 auto v3 = torch::distributed::rpc::cloneSparseTensors({sparse}); in TEST()
64 TORCH_API c10::List<at::Tensor> cloneSparseTensors(
321 c10::List<at::Tensor> cloneSparseTensors( in cloneSparseTensors() function376 pickler.pushIValue(cloneSparseTensors(tensors)); in wireSerialize()
164 buffers.tensors = cloneSparseTensors(rpcMessage->tensors()).vec(); in tensorpipeSerialize()