Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_flatten.h24 inline std::vector<at::Tensor> unflatten_dense_tensors( in unflatten_dense_tensors() function
27 return at::unflatten_dense_tensors(flat, tensors); in unflatten_dense_tensors()
H A Dtensor_flatten.cpp110 utils::unflatten_dense_tensors(flat_indices, fmap(tensors, &get_indices)); in unflatten_sparse_tensors()
112 utils::unflatten_dense_tensors(flat_values, fmap(tensors, &get_values)); in unflatten_sparse_tensors()
/aosp_15_r20/external/pytorch/torch/
H A D_utils.py555 return torch._C._nn.unflatten_dense_tensors(flat, tensors)
574 indices = torch._C._nn.unflatten_dense_tensors(
577 values = torch._C._nn.unflatten_dense_tensors(
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtensor_flatten.cpp19 torch::utils::unflatten_dense_tensors(flatTensor, tensorList); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dcomm.cpp30 auto output_tensors = torch::utils::unflatten_dense_tensors( in finish()
/aosp_15_r20/external/pytorch/torch/_C/
H A D_nn.pyi.in70 def unflatten_dense_tensors(flat: Tensor, tensors: List[Tensor]) -> List[Tensor]: ...
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp210 torch::utils::unflatten_dense_tensors(results[i], chunk.tensors)) { in _broadcast_out_impl()
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dannotated_ops557 unflatten_dense_tensors, misc
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp3948 std::vector<Tensor> unflatten_dense_tensors(const Tensor& flat, TensorList tensors) { in unflatten_dense_tensors() function