Searched refs:unflatten_dense_tensors (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/pytorch/torch/csrc/utils/ |
H A D | tensor_flatten.h | 24 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 D | tensor_flatten.cpp | 110 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.py | 555 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 D | tensor_flatten.cpp | 19 torch::utils::unflatten_dense_tensors(flatTensor, tensorList); in TEST()
|
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
H A D | comm.cpp | 30 auto output_tensors = torch::utils::unflatten_dense_tensors( in finish()
|
/aosp_15_r20/external/pytorch/torch/_C/ |
H A D | _nn.pyi.in | 70 def unflatten_dense_tensors(flat: Tensor, tensors: List[Tensor]) -> List[Tensor]: ...
|
/aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
H A D | comm.cpp | 210 torch::utils::unflatten_dense_tensors(results[i], chunk.tensors)) { in _broadcast_out_impl()
|
/aosp_15_r20/external/pytorch/functorch/op_analysis/ |
H A D | annotated_ops | 557 unflatten_dense_tensors, misc
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | TensorShape.cpp | 3948 std::vector<Tensor> unflatten_dense_tensors(const Tensor& flat, TensorList tensors) { in unflatten_dense_tensors() function
|