Searched refs:AddReduce (Results 1 – 3 of 3) sorted by relevance
254 llvm::Value* VectorSupportLibrary::AddReduce(llvm::Value* vector) { in AddReduce() function in xla::cpu::VectorSupportLibrary349 [this](llvm::Value* vector) { return AddReduce(vector); }); in ComputeHorizontalSums()
243 llvm::Value* AddReduce(llvm::Value* vector);
314 HloInstruction* AddReduce(HloInstruction* hlo, absl::Span<const int64> dims) { in AddReduce() function in xla::__anon4eef97bd0111::AlgebraicSimplifierVisitor326 HloInstruction* AddReduce(HloInstruction* hlo, int64 dim) { in AddReduce() function in xla::__anon4eef97bd0111::AlgebraicSimplifierVisitor327 return AddReduce(hlo, std::vector<int64>{dim}); in AddReduce()1187 return AddReduce(AsType(hlo, F32), dim); in HandleDotStrengthReduction()1759 new_dot = AddReduce(new_dot, reduce_dims); in HandleDot()