Searched defs:dims_to_reduce (Results 1 – 2 of 2) sorted by relevance
117 Eigen::DSizes<Eigen::DenseIndex, 1> dims_to_reduce; in Compute() local120 Eigen::IndexList<Eigen::type2index<0> > dims_to_reduce; in Compute() local
3493 const Shape& input_shape, absl::Span<const int64> dims_to_reduce) { in GetDimensionsToKeepMinorToMajor()3525 const Shape& input_shape, absl::Span<const int64> dims_to_reduce) { in GetReductionToVectorDimensions()