Searched refs:dims_to_reduce (Results 1 – 2 of 2) sorted by relevance
117 Eigen::DSizes<Eigen::DenseIndex, 1> dims_to_reduce; in Compute() local118 dims_to_reduce[0] = 0; in Compute()120 Eigen::IndexList<Eigen::type2index<0> > dims_to_reduce; in Compute() local188 out_slice = in_slice.reduce(dims_to_reduce, Reducer()); in Compute()
3493 const Shape& input_shape, absl::Span<const int64> dims_to_reduce) { in GetDimensionsToKeepMinorToMajor() argument3498 auto it = absl::c_find_if(dims_to_reduce, [&](int64 dim_to_reduce) { in GetDimensionsToKeepMinorToMajor()3501 if (it == dims_to_reduce.end()) { in GetDimensionsToKeepMinorToMajor()3525 const Shape& input_shape, absl::Span<const int64> dims_to_reduce) { in GetReductionToVectorDimensions() argument3527 GetDimensionsToKeepMinorToMajor(input_shape, dims_to_reduce); in GetReductionToVectorDimensions()