Searched refs:grad_reduce_idx_ (Results 1 – 1 of 1) sorted by relevance
98 const Vec& grad_reduce_idx(int i) const { return grad_reduce_idx_[i]; } in grad_reduce_idx()118 Vec grad_reduce_idx_[N]; variable238 grad_reduce_idx_[i].push_back(largest_rank - 1 - j); in BCastList()268 grad_reduce_idx_[i].push_back(largest_rank - 1 - j); in BCastList()277 grad_reduce_idx_[i].push_back(largest_rank - 1 - j); in BCastList()297 Reverse(&grad_reduce_idx_[i]); in BCastList()380 const Vec& grad_x_reduce_idx() const { return grad_reduce_idx_[0]; } in grad_x_reduce_idx()381 const Vec& grad_y_reduce_idx() const { return grad_reduce_idx_[1]; } in grad_y_reduce_idx()