Searched refs:dimension_to_reduce (Results 1 – 1 of 1) sorted by relevance
131 int dimension_to_reduce, T init, in ReduceR2ToR1() argument133 std::vector<T> result(dimension_to_reduce == 0 ? input.n2() : input.n1(), in ReduceR2ToR1()137 int output = dimension_to_reduce == 0 ? i1 : i0; in ReduceR2ToR1()