Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/c10d/
DProcessGroupGloo.cpp1149 void populate_from_sparse_tensor(const at::Tensor& tensor) { in populate_from_sparse_tensor() function in c10d::__anon725d3f3f0911::AsyncSparseAllreduceWork::SparseTensorMetadata
1275 metadata[context->rank].populate_from_sparse_tensor(tensor); in allgather_metadata()