Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_flatten.h76 TORCH_API std::pair<at::Tensor, at::Tensor> flatten_sparse_tensors(
H A Dtensor_flatten.cpp95 std::pair<at::Tensor, at::Tensor> flatten_sparse_tensors( in flatten_sparse_tensors() function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp188 auto flat_tuple = torch::utils::flatten_sparse_tensors(chunk.tensors); in _broadcast_out_impl()