Searched refs:grad_x_reduce_idx_ (Results 1 – 2 of 2) sorted by relevance
88 grad_x_reduce_idx_.push_back(n - 1 - i); in BCast()107 grad_x_reduce_idx_.push_back(n - 1 - i); in BCast()153 Reverse(&grad_x_reduce_idx_); in BCast()
101 const Vec& grad_x_reduce_idx() const { return grad_x_reduce_idx_; } in grad_x_reduce_idx()131 Vec grad_x_reduce_idx_; variable