Searched refs:grad_y_reduce_idx_ (Results 1 – 2 of 2) sorted by relevance
95 grad_y_reduce_idx_.push_back(n - 1 - i); in BCast()108 grad_y_reduce_idx_.push_back(n - 1 - i); in BCast()154 Reverse(&grad_y_reduce_idx_); in BCast()
102 const Vec& grad_y_reduce_idx() const { return grad_y_reduce_idx_; } in grad_y_reduce_idx()132 Vec grad_y_reduce_idx_; variable