Searched refs:compute_bucket_assignment_by_size (Results 1 – 4 of 4) sorted by relevance
71 compute_bucket_assignment_by_size(tensors.vec(), {buffer_size}); in broadcast_coalesced()
574 compute_bucket_assignment_by_size(
1837 compute_bucket_assignment_by_size( in rebuild_buckets()2094 compute_bucket_assignment_by_size( in compute_bucket_assignment_by_size() function
3059 return ::c10d::compute_bucket_assignment_by_size( in c10d_init()3066 return ::c10d::compute_bucket_assignment_by_size( in c10d_init()