Searched refs:flatten_dense_tensors (Results 1 – 8 of 8) sorted by relevance
20 inline at::Tensor flatten_dense_tensors(at::TensorList tensors) { in flatten_dense_tensors() function21 return at::flatten_dense_tensors(tensors); in flatten_dense_tensors()
97 auto flat_indices = utils::flatten_dense_tensors(fmap(tensors, &get_indices)); in flatten_sparse_tensors()98 auto flat_values = utils::flatten_dense_tensors(fmap(tensors, &get_values)); in flatten_sparse_tensors()
519 return torch._C._nn.flatten_dense_tensors(tensors)533 flat_indices = torch._C._nn.flatten_dense_tensors(536 flat_values = torch._C._nn.flatten_dense_tensors(
20 flat_tensor_({torch::utils::flatten_dense_tensors(bucket_tensors_)}) { in BroadcastWork()
69 def flatten_dense_tensors(tensors: List[Tensor]) -> Tensor: ...
205 torch::utils::flatten_dense_tensors(chunk.tensors), devices); in _broadcast_out_impl()
556 flatten_dense_tensors, misc
3941 Tensor flatten_dense_tensors(TensorList tensors) { in flatten_dense_tensors() function