Home
last modified time | relevance | path

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

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