Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_flatten.h79 TORCH_API std::vector<at::Tensor> unflatten_sparse_tensors(
H A Dtensor_flatten.cpp102 std::vector<at::Tensor> unflatten_sparse_tensors( in unflatten_sparse_tensors() function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp197 for (const auto& var : torch::utils::unflatten_sparse_tensors( in _broadcast_out_impl()