Searched defs:SparseTensorMetadata (Results 1 – 1 of 1) sorted by relevance
1134 class SparseTensorMetadata { class in c10d::__anon725d3f3f0911::AsyncSparseAllreduceWork1140 explicit SparseTensorMetadata(at::Tensor metadata) in SparseTensorMetadata() function in c10d::__anon725d3f3f0911::AsyncSparseAllreduceWork::SparseTensorMetadata