Searched refs:reduction_loops (Results 1 – 1 of 1) sorted by relevance
234 std::vector<mlir::AffineForOp> reduction_loops; member273 std::vector<mlir::AffineForOp> reduction_loops; in CreateNaiveMlirConv() local274 reduction_loops = CreateNestedSimpleLoops( in CreateNaiveMlirConv()279 mlir::AffineForOp loop_c = reduction_loops[0]; in CreateNaiveMlirConv()286 for (auto loop : absl::MakeSpan(reduction_loops).subspan(1)) { in CreateNaiveMlirConv()292 builder = OpBuilder::atBlockTerminator(reduction_loops.back().getBody()); in CreateNaiveMlirConv()353 builder.setInsertionPointAfter(reduction_loops[0]); in CreateNaiveMlirConv()369 return InitialMlirConvAnchors{cartesian_product_loops, reduction_loops, in CreateNaiveMlirConv()392 std::vector<mlir::AffineForOp> reduction_loops; member399 std::vector<mlir::AffineForOp> reduction_loops = anchors.reduction_loops; in TransformMlirConv() local[all …]