Searched defs:dims_to_reduce (Results 1 – 5 of 5) sorted by relevance
125 Eigen::DSizes<Eigen::DenseIndex, 1> dims_to_reduce; in Compute() local128 Eigen::IndexList<Eigen::type2index<0> > dims_to_reduce; in Compute() local
119 SmallVector<int64_t, 4> dims_to_reduce; in runOnFunction() local
164 std::vector<int64> dims_to_reduce = {inner_reduced_dimension}; in RewriteReduction() local
216 const Shape& input_shape, absl::Span<const int64> dims_to_reduce) { in GetReductionKindAndContiguousComponentsImpl()
151 llvm::SmallVector<int64_t, 4> dims_to_reduce(rank - 1); in GetBiasAddGradReductionIndices() local