Searched refs:reduce_on_cols (Results 1 – 1 of 1) sorted by relevance
114 typename Eigen::IndexList<Eigen::type2index<1>> reduce_on_cols; in Compute() local118 Eigen::array<int, 1> reduce_on_cols = {1}; in Compute()123 input.maximum(/*dims=*/reduce_on_cols).eval().reshape(rows_by_one); in Compute()