Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/c10d/
Dreducer.cpp1772 std::vector<at::Tensor> bucket_sizes_tensor_list = { in sync_bucket_indices() local
1774 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()