Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp1770 auto bucket_sizes_tensor_device = at::empty({(int64_t)num_buckets}, options); in sync_bucket_indices() local
1771 bucket_sizes_tensor_device.copy_(bucket_sizes_tensor, /*non_blocking=*/true); in sync_bucket_indices()
1773 bucket_sizes_tensor_device}; in sync_bucket_indices()