Searched refs:flat_tensor_ (Results 1 – 1 of 1) sorted by relevance
20 flat_tensor_({torch::utils::flatten_dense_tensors(bucket_tensors_)}) { in BroadcastWork()23 work_ = process_group->broadcast(flat_tensor_, broadcastOptions);31 flat_tensor_.front(), bucket_tensors_); in finish()52 std::vector<at::Tensor> flat_tensor_; member in c10d::__anon1c46c9ee0111::BroadcastWork