Searched refs:rows_by_one (Results 1 – 1 of 1) sorted by relevance
115 typename Eigen::IndexList<int, Eigen::type2index<1>> rows_by_one; in Compute() local116 rows_by_one.set(0, num_rows); in Compute()119 Eigen::array<int, 2> rows_by_one = {static_cast<int>(num_rows), 1}; in Compute()123 input.maximum(/*dims=*/reduce_on_cols).eval().reshape(rows_by_one); in Compute()