Searched refs:bucket_sizes_tensor_list (Results 1 – 1 of 1) sorted by relevance
1772 std::vector<at::Tensor> bucket_sizes_tensor_list = { in sync_bucket_indices() local1774 process_group_->broadcast(bucket_sizes_tensor_list)->wait(); in sync_bucket_indices()1776 bucket_sizes_tensor_list.front(), /*non_blocking=*/false); in sync_bucket_indices()