Searched defs:reductions (Results 1 – 3 of 3) sorted by relevance
63 std::vector<int64> sizes_in_kib, std::vector<HloComputation*> reductions, in MakeCrossReplicaReductions()
2498 ArrayRef<AtomicRMWKind> reductions, in build()2513 ArrayRef<AtomicRMWKind> reductions, in build()2528 ArrayRef<AtomicRMWKind> reductions, in build()2836 SmallVector<Attribute, 4> reductions; in parseAffineParallelOp() local
891 SmallVector<ReduceOp, 4> reductions(body->getOps<ReduceOp>()); in verify() local