Home
last modified time | relevance | path

Searched defs:compressed_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Ddefault_comm_hooks.cpp21 auto compressed_tensor = bucket.getBufferRef().to(torch::kFloat16); in runHook() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DcuSPARSELtOps.cpp62 auto compressed_tensor = sparse_input.new_empty(sparse_input.numel() * compression_factor / 16); in _cslt_compress() local