Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp1762 auto bucket_sizes_tensor = at::empty({(int64_t)num_buckets}, at::kInt); in sync_bucket_indices() local
1763 auto bucket_sizes_accessor = bucket_sizes_tensor.accessor<int, 1>(); in sync_bucket_indices()
1771 bucket_sizes_tensor_device.copy_(bucket_sizes_tensor, /*non_blocking=*/true); in sync_bucket_indices()
1775 bucket_sizes_tensor.copy_( in sync_bucket_indices()