Searched defs:dims_to_reduce (Results 1 – 7 of 7) sorted by relevance
44 ArrayRef<int32_t> dims_to_reduce, in GetReductionIR()64 llvm::ArrayRef<int32_t> dims_to_reduce) { in GetSumF32IR()71 llvm::ArrayRef<int32_t> dims_to_reduce) { in GetMeanF32IR()
36 std::array<int32_t, N> dims_to_reduce) { in ReducedTensorShape()51 std::array<int32_t, N_DIMS_TO_REDUCE> dims_to_reduce) { in GetEigenSumF32Function()69 std::array<int32_t, N_DIMS_TO_REDUCE> dims_to_reduce) { in GetEigenMeanF32Function()
59 std::array<int64_t, 1> dims_to_reduce{0}; in EigenFusedReduction1D() local
128 Eigen::IndexList<Eigen::type2index<0> > dims_to_reduce; in Compute() local
187 std::vector<int64_t> dims_to_reduce = {inner_reduced_dimension}; in RewriteReduction() local
155 const Shape& input_shape, absl::Span<const int64_t> dims_to_reduce) { in GetReductionKindAndContiguousComponentsImpl()223 const Shape& operand_shape, absl::Span<int64_t const> dims_to_reduce) { in IsReductionFromOrToContiguousDimensionsImpl()
141 llvm::SmallVector<int64_t, 4> dims_to_reduce(rank - 1); in GetBiasAddGradReductionIndices() local