Searched refs:align_size_list_ (Results 1 – 4 of 4) sorted by relevance
49 (void)align_size_list_.emplace_back(align_size); in AllocateAllReduceAddr()65 if (origin_size_list.size() < bucket_size_ || align_size_list_.size() < bucket_size_) { in AllocateAllReduceAddr()67 << " align_size_list.size:" << align_size_list_.size(); in AllocateAllReduceAddr()71 memcpy_output += align_size_list_[i]; in AllocateAllReduceAddr()
114 tensor_output += align_size_list_[i]; in UpdateTensorOutputAddr()129 align_size_list_.clear(); in Release()
71 std::vector<size_t> align_size_list_; variable
64 align_size_list_.emplace_back(align_size); in AllocateAllReduceAddr()77 memcpy_output += align_size_list_[i]; in AllocateAllReduceAddr()